File size: 1,259 Bytes
10830cf 429dbbe 10830cf 429dbbe 10830cf 429dbbe 10830cf 429dbbe 10830cf 429dbbe 10830cf 429dbbe 10830cf c6eaaf8 |
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 |
{
"labels": [
0,
1,
2,
3,
4
],
"num_labels": 5,
"environment": {
"python": "3.11.14 (main, Oct 9 2025, 16:16:55) [Clang 17.0.0 (clang-1700.0.13.3)]",
"scikit-learn": "1.8.0",
"setfit": "1.1.3",
"torch": "2.9.1"
},
"base_model": "BAAI/bge-base-en-v1.5",
"serialization": "safetensors",
"training_config": {
"batch_size": 4,
"num_epochs": [
1,
16
],
"training_samples": 540,
"test_samples": 100
},
"model_stats": {
"total_parameters": 109482240,
"trainable_parameters": 109482240,
"body_parameters": 109482240,
"head_parameters": 0,
"model_size_mb": 417.64,
"model_description": "SetFit model with 109,482,240 total parameters (417.6 MB)"
},
"evaluation_metrics": {
"accuracy": 0.9,
"f1_weighted": 0.8984430773904458,
"f1_macro": 0.8984430773904458,
"precision_weighted": 0.9060606060606061,
"precision_macro": 0.906060606060606,
"recall_weighted": 0.9,
"recall_macro": 0.9
},
"training_summary": {
"epochs_trained": 5,
"final_loss": 0.0648
}
} |