model_metadata.json

{
    "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
    }
}