loganh274 commited on
Commit
9652d2f
·
verified ·
1 Parent(s): 5a752fd

Upload model_metadata.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_metadata.json +10 -10
model_metadata.json CHANGED
@@ -8,7 +8,7 @@
8
  ],
9
  "num_labels": 5,
10
  "environment": {
11
- "python": "3.11.14 (main, Oct 9 2025, 16:16:55) [Clang 17.0.0 (clang-1700.0.13.3)]",
12
  "scikit-learn": "1.8.0",
13
  "setfit": "1.1.3",
14
  "torch": "2.9.1"
@@ -16,7 +16,7 @@
16
  "base_model": "BAAI/bge-base-en-v1.5",
17
  "serialization": "safetensors",
18
  "training_config": {
19
- "batch_size": 4,
20
  "num_epochs": [
21
  1,
22
  16
@@ -33,16 +33,16 @@
33
  "model_description": "SetFit model with 109,482,240 total parameters (417.6 MB)"
34
  },
35
  "evaluation_metrics": {
36
- "accuracy": 0.9,
37
- "f1_weighted": 0.8984430773904458,
38
- "f1_macro": 0.8984430773904458,
39
- "precision_weighted": 0.9060606060606061,
40
- "precision_macro": 0.906060606060606,
41
- "recall_weighted": 0.9,
42
- "recall_macro": 0.9
43
  },
44
  "training_summary": {
45
  "epochs_trained": 5,
46
- "final_loss": 0.0648
47
  }
48
  }
 
8
  ],
9
  "num_labels": 5,
10
  "environment": {
11
+ "python": "3.11.14 (main, Oct 9 2025, 16:16:55) [Clang 17.0.0 (clang-1700.4.4.1)]",
12
  "scikit-learn": "1.8.0",
13
  "setfit": "1.1.3",
14
  "torch": "2.9.1"
 
16
  "base_model": "BAAI/bge-base-en-v1.5",
17
  "serialization": "safetensors",
18
  "training_config": {
19
+ "batch_size": 16,
20
  "num_epochs": [
21
  1,
22
  16
 
33
  "model_description": "SetFit model with 109,482,240 total parameters (417.6 MB)"
34
  },
35
  "evaluation_metrics": {
36
+ "accuracy": 0.88,
37
+ "f1_weighted": 0.8805050505050506,
38
+ "f1_macro": 0.8805050505050506,
39
+ "precision_weighted": 0.8883333333333333,
40
+ "precision_macro": 0.8883333333333333,
41
+ "recall_weighted": 0.88,
42
+ "recall_macro": 0.8799999999999999
43
  },
44
  "training_summary": {
45
  "epochs_trained": 5,
46
+ "final_loss": 0.0893
47
  }
48
  }