英文

sd-ner-v2

这个模型是在源数据NLP数据集上对 microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract 进行微调得到的。在评估集上达到以下结果:

  • 损失:0.1551
  • 准确率:0.9513
  • 精确度:0.8030
  • 召回率:0.8378
  • F1值:0.8200

模型描述

需要更多信息

预期用途和限制

需要更多信息

训练和评估数据

需要更多信息

训练过程

训练超参数

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

  • 学习率:0.0001
  • 训练批大小:64
  • 评估批大小:256
  • 种子:42
  • 优化器:Adafactor
  • 学习率调度器类型:线性
  • 训练轮数:2.0

训练结果

Training Loss Epoch Step Validation Loss Accuracy Score Precision Recall F1
0.1082 1.0 785 0.1550 0.9493 0.7826 0.8402 0.8104
0.073 2.0 1570 0.1551 0.9513 0.8030 0.8378 0.8200

框架版本

  • Transformers 4.20.0
  • Pytorch 1.11.0a0+bfe5ad2
  • Datasets 1.17.0
  • Tokenizers 0.12.1