| { | |
| "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.4.4.1)]", | |
| "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": 16, | |
| "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.88, | |
| "f1_weighted": 0.8805050505050506, | |
| "f1_macro": 0.8805050505050506, | |
| "precision_weighted": 0.8883333333333333, | |
| "precision_macro": 0.8883333333333333, | |
| "recall_weighted": 0.88, | |
| "recall_macro": 0.8799999999999999 | |
| }, | |
| "training_summary": { | |
| "epochs_trained": 5, | |
| "final_loss": 0.0893 | |
| } | |
| } |