模型:

daveni/aesthetic_attribute_classifier

英文

美学属性分类器

该模型是在 PCCD dataset 上经过微调的版本。它在评估集上取得以下结果:

  • 损失:0.3976
  • 精确度:{'precision': 0.877129341279301}
  • 召回率:{'recall': 0.8751381215469614}
  • F1 值:{'f1': 0.875529982855803}
  • 准确度:{'accuracy': 0.8751381215469614}

模型描述

需要更多信息

预期用途和限制

需要更多信息

训练和评估数据

需要更多信息

训练过程

训练超参数

训练时使用了以下超参数:

  • 学习率:2e-05
  • 训练批大小:16
  • 评估批大小:16
  • 种子:42
  • 优化器:Adam,参数为 betas=(0.9,0.999) 和 epsilon=1e-08
  • lr_scheduler_type: linear
  • epochs 数量:5

训练结果

Training Loss Epoch Step Validation Loss Precision Recall F1 Accuracy
0.452 1.0 1528 0.4109 {'precision': 0.8632779077963935} {'recall': 0.8615101289134438} {'f1': 0.8618616182904953} {'accuracy': 0.8615101289134438}
0.3099 2.0 3056 0.3976 {'precision': 0.877129341279301} {'recall': 0.8751381215469614} {'f1': 0.875529982855803} {'accuracy': 0.8751381215469614}
0.227 3.0 4584 0.4320 {'precision': 0.876211408446225} {'recall': 0.874401473296501} {'f1': 0.8747427955387239} {'accuracy': 0.874401473296501}
0.1645 4.0 6112 0.4840 {'precision': 0.8724641667216837} {'recall': 0.8714548802946593} {'f1': 0.8714577820909117} {'accuracy': 0.8714548802946593}
0.1141 5.0 7640 0.5083 {'precision': 0.8755445355051571} {'recall': 0.8747697974217311} {'f1': 0.8748766125899489} {'accuracy': 0.8747697974217311}

框架版本

  • Transformers 4.16.2
  • Pytorch 1.10.2+cu113
  • Datasets 1.18.3
  • Tokenizers 0.11.0