Modelpublic

slepape/ArticleTypePrediction

sourceHugging Facemitupdated 1y agoView on Hugging Face

0likes14downloads

training_config.json35 linesDownload Raw Back to root

1{2  "model_name": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",3  "task": "text-classification",4  "num_labels": 9,5  "classes": [6    "CASE_REPORTS",7    "COMMENT",8    "EDITORIAL",9    "GUIDELINES",10    "META_ANALYSIS",11    "PROSPECTIVE",12    "RCT",13    "RETROSPECTIVE",14    "REVIEW"15  ],16  "training_details": {17    "final_accuracy": 0.937962962962963,18    "final_f1": 0.9377,19    "total_samples": 54000,20    "epochs": 10,21    "learning_rate": 0.0001,22    "batch_size": 32,23    "max_length": 512,24    "best_checkpoint": "checkpoint-2000"25  },26  "text_preprocessing": {27    "format": "[TITLE] {title} [SEP] [ABSTRACT] {abstract}",28    "special_tokens": [29      "[TITLE]",30      "[SEP]",31      "[ABSTRACT]"32    ],33    "truncation_strategy": "optimized_title_abstract_split"34  }35}