File size: 1,688 Bytes
9bcdf02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
{
"model_info": {
"model_type": "distilroberta-base",
"task": "binary_classification",
"classes": [
"human",
"bot"
],
"num_parameters": "82M",
"framework": "transformers",
"pytorch_version": ">=1.12.0"
},
"training_methodology": {
"method": "class_weighted_cross_validation",
"cv_folds": 5,
"cv_strategy": "stratified",
"early_stopping": true,
"early_stopping_patience": 3,
"mixed_precision": "fp16"
},
"hyperparameters": {
"batch_size_per_device": 2,
"gradient_accumulation_steps": 8,
"max_epochs": 10,
"weight_decay": 0.01,
"optimizer": "AdamW"
},
"performance_summary": {
"cv_metrics": {
"accuracy": {
"mean": 0.9433,
"std": 0.0052,
"min": 0.9385,
"max": 0.9497
},
"f1_weighted": {
"mean": 0.9434,
"std": 0.0051,
"min": 0.9387,
"max": 0.9497
},
"f1_macro": {
"mean": 0.9419,
"std": 0.0052,
"min": 0.9371,
"max": 0.9483
},
"precision_weighted": {
"mean": 0.9444,
"std": 0.0045,
"min": 0.9397,
"max": 0.9498
},
"recall_weighted": {
"mean": 0.9433,
"std": 0.0052,
"min": 0.9385,
"max": 0.9497
}
},
"test_metrics": {
"loss": 0.1511,
"accuracy": 0.9423,
"precision_weighted": 0.9428,
"recall_weighted": 0.9423,
"f1_weighted": 0.9424,
"precision_macro": 0.9393,
"recall_macro": 0.9427,
"f1_macro": 0.9409,
"runtime": 121.6927,
"samples_per_second": 232.832,
"steps_per_second": 8.316
}
}
} |