lgsilvaesilva commited on
Commit
9904d93
·
verified ·
1 Parent(s): 1d64bbe

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +4 -0
  2. README.md +326 -0
  3. REPORT.md +160 -0
  4. baselines/embedding-lightgbm/embedding-lightgbm.joblib +3 -0
  5. baselines/embedding-lightgbm/test_predictions.csv +0 -0
  6. baselines/embedding-lightgbm/validation_predictions.csv +0 -0
  7. baselines/embedding-logistic/embedding-logistic.joblib +3 -0
  8. baselines/embedding-logistic/test_predictions.csv +0 -0
  9. baselines/embedding-logistic/validation_predictions.csv +0 -0
  10. baselines/embedding-svm/embedding-svm.joblib +3 -0
  11. baselines/embedding-svm/test_predictions.csv +0 -0
  12. baselines/embedding-svm/validation_predictions.csv +0 -0
  13. baselines/logistic/logistic_tfidf.joblib +3 -0
  14. baselines/logistic/test_predictions.csv +0 -0
  15. baselines/logistic/validation_predictions.csv +0 -0
  16. baselines/xgboost/test_predictions.csv +0 -0
  17. baselines/xgboost/validation_predictions.csv +0 -0
  18. baselines/xgboost/xgboost_tfidf.joblib +3 -0
  19. report.json +1257 -0
  20. transformer/checkpoint-296/config.json +39 -0
  21. transformer/checkpoint-296/model.safetensors +3 -0
  22. transformer/checkpoint-296/optimizer.pt +3 -0
  23. transformer/checkpoint-296/rng_state.pth +3 -0
  24. transformer/checkpoint-296/scaler.pt +3 -0
  25. transformer/checkpoint-296/scheduler.pt +3 -0
  26. transformer/checkpoint-296/tokenizer.json +3 -0
  27. transformer/checkpoint-296/tokenizer_config.json +15 -0
  28. transformer/checkpoint-296/trainer_state.json +133 -0
  29. transformer/checkpoint-296/training_args.bin +3 -0
  30. transformer/checkpoint-592/config.json +39 -0
  31. transformer/checkpoint-592/model.safetensors +3 -0
  32. transformer/checkpoint-592/optimizer.pt +3 -0
  33. transformer/checkpoint-592/rng_state.pth +3 -0
  34. transformer/checkpoint-592/scaler.pt +3 -0
  35. transformer/checkpoint-592/scheduler.pt +3 -0
  36. transformer/checkpoint-592/tokenizer.json +3 -0
  37. transformer/checkpoint-592/tokenizer_config.json +15 -0
  38. transformer/checkpoint-592/trainer_state.json +230 -0
  39. transformer/checkpoint-592/training_args.bin +3 -0
  40. transformer/checkpoint-888/config.json +39 -0
  41. transformer/checkpoint-888/model.safetensors +3 -0
  42. transformer/checkpoint-888/optimizer.pt +3 -0
  43. transformer/checkpoint-888/rng_state.pth +3 -0
  44. transformer/checkpoint-888/scaler.pt +3 -0
  45. transformer/checkpoint-888/scheduler.pt +3 -0
  46. transformer/checkpoint-888/tokenizer.json +3 -0
  47. transformer/checkpoint-888/tokenizer_config.json +15 -0
  48. transformer/checkpoint-888/trainer_state.json +327 -0
  49. transformer/checkpoint-888/training_args.bin +3 -0
  50. transformer/config.json +46 -0
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ transformer/checkpoint-296/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ transformer/checkpoint-592/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ transformer/checkpoint-888/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
+ transformer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,326 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ pipeline_tag: text-classification
4
+ base_model: FacebookAI/xlm-roberta-base
5
+ tags:
6
+ - text-classification
7
+ - binary-classification
8
+ - amis
9
+ - agriculture
10
+ language: multilingual
11
+ ---
12
+
13
+ # AMIS Commodity Classifier
14
+
15
+ This model repository contains artifacts from an AMIS commodity relevance classifier training run.
16
+ It includes the Transformer model, any configured TF-IDF or sentence-embedding baselines, prediction files, and the training report.
17
+
18
+ - Dataset: `faodl/amis-agri-maize_corn`
19
+ - Dataset subset: ``
20
+ - Dataset revision: `main`
21
+ - Text column: `chunk_text`
22
+ - Label column: `label`
23
+ - Transformer: `FacebookAI/xlm-roberta-base`
24
+ - Generated at: `2026-06-05T20:32:51.106165+00:00`
25
+
26
+ ## Dataset Summary
27
+
28
+ | Split | Rows | Label 0 | Label 1 | Unique groups | Mean text length |
29
+ | --- | ---: | ---: | ---: | ---: | ---: |
30
+ | train | 4724 | 3822 | 902 | 2226 | 702.9 |
31
+ | validation | 1060 | 843 | 217 | 477 | 708.3 |
32
+ | test | 1054 | 819 | 235 | 478 | 711.9 |
33
+
34
+ ## Threshold Comparison on Validation Split
35
+
36
+ Validation metrics document threshold selection and tuning behavior; test metrics remain the primary estimate of out-of-sample performance.
37
+
38
+ | Model | Threshold | Accuracy | Precision | Recall | F1 | ROC AUC | Average precision |
39
+ | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
40
+ | logistic_tfidf | 0.500 | 0.896 | 0.700 | 0.862 | 0.773 | 0.929 | 0.841 |
41
+ | logistic_tfidf | 0.586 | 0.915 | 0.777 | 0.820 | 0.798 | 0.929 | 0.841 |
42
+ | xgboost_tfidf | 0.500 | 0.957 | 0.913 | 0.871 | 0.892 | 0.967 | 0.914 |
43
+ | xgboost_tfidf | 0.379 | 0.958 | 0.902 | 0.894 | 0.898 | 0.967 | 0.914 |
44
+ | embedding-logistic_sentence_embeddings | 0.500 | 0.881 | 0.649 | 0.912 | 0.759 | 0.959 | 0.867 |
45
+ | embedding-logistic_sentence_embeddings | 0.744 | 0.923 | 0.808 | 0.816 | 0.812 | 0.959 | 0.867 |
46
+ | embedding-svm_sentence_embeddings | 0.500 | 0.913 | 0.849 | 0.700 | 0.768 | 0.955 | 0.858 |
47
+ | embedding-svm_sentence_embeddings | 0.401 | 0.914 | 0.789 | 0.793 | 0.791 | 0.955 | 0.858 |
48
+ | embedding-lightgbm_sentence_embeddings | 0.500 | 0.916 | 0.791 | 0.802 | 0.796 | 0.963 | 0.878 |
49
+ | embedding-lightgbm_sentence_embeddings | 0.145 | 0.916 | 0.746 | 0.894 | 0.813 | 0.963 | 0.878 |
50
+ | transformer | 0.500 | 0.958 | 0.913 | 0.876 | 0.894 | 0.973 | 0.943 |
51
+ | transformer | 0.328 | 0.959 | 0.907 | 0.894 | 0.900 | 0.973 | 0.943 |
52
+
53
+ ## Threshold Comparison on Test Split
54
+
55
+ | Model | Threshold | Accuracy | Precision | Recall | F1 | ROC AUC | Average precision |
56
+ | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
57
+ | logistic_tfidf | 0.500 | 0.910 | 0.787 | 0.817 | 0.802 | 0.953 | 0.863 |
58
+ | logistic_tfidf | 0.586 | 0.915 | 0.857 | 0.740 | 0.795 | 0.953 | 0.863 |
59
+ | xgboost_tfidf | 0.500 | 0.942 | 0.914 | 0.817 | 0.863 | 0.968 | 0.920 |
60
+ | xgboost_tfidf | 0.379 | 0.948 | 0.895 | 0.868 | 0.881 | 0.968 | 0.920 |
61
+ | embedding-logistic_sentence_embeddings | 0.500 | 0.884 | 0.704 | 0.830 | 0.762 | 0.936 | 0.844 |
62
+ | embedding-logistic_sentence_embeddings | 0.744 | 0.896 | 0.831 | 0.668 | 0.741 | 0.936 | 0.844 |
63
+ | embedding-svm_sentence_embeddings | 0.500 | 0.894 | 0.892 | 0.596 | 0.714 | 0.932 | 0.842 |
64
+ | embedding-svm_sentence_embeddings | 0.401 | 0.902 | 0.851 | 0.681 | 0.757 | 0.932 | 0.842 |
65
+ | embedding-lightgbm_sentence_embeddings | 0.500 | 0.907 | 0.870 | 0.685 | 0.767 | 0.953 | 0.873 |
66
+ | embedding-lightgbm_sentence_embeddings | 0.145 | 0.901 | 0.784 | 0.770 | 0.777 | 0.953 | 0.873 |
67
+ | transformer | 0.500 | 0.935 | 0.892 | 0.804 | 0.846 | 0.953 | 0.890 |
68
+ | transformer | 0.328 | 0.935 | 0.881 | 0.817 | 0.848 | 0.953 | 0.890 |
69
+
70
+ ## Confusion Matrices on Test Split
71
+
72
+ Rows are true labels and columns are predicted labels.
73
+
74
+ ### logistic_tfidf at threshold 0.500
75
+
76
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
77
+ | --- | ---: | ---: |
78
+ | NOT_RELEVANT | 767 | 52 |
79
+ | RELEVANT | 43 | 192 |
80
+
81
+ ### logistic_tfidf at threshold 0.586
82
+
83
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
84
+ | --- | ---: | ---: |
85
+ | NOT_RELEVANT | 790 | 29 |
86
+ | RELEVANT | 61 | 174 |
87
+
88
+ ### xgboost_tfidf at threshold 0.500
89
+
90
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
91
+ | --- | ---: | ---: |
92
+ | NOT_RELEVANT | 801 | 18 |
93
+ | RELEVANT | 43 | 192 |
94
+
95
+ ### xgboost_tfidf at threshold 0.379
96
+
97
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
98
+ | --- | ---: | ---: |
99
+ | NOT_RELEVANT | 795 | 24 |
100
+ | RELEVANT | 31 | 204 |
101
+
102
+ ### embedding-logistic_sentence_embeddings at threshold 0.500
103
+
104
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
105
+ | --- | ---: | ---: |
106
+ | NOT_RELEVANT | 737 | 82 |
107
+ | RELEVANT | 40 | 195 |
108
+
109
+ ### embedding-logistic_sentence_embeddings at threshold 0.744
110
+
111
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
112
+ | --- | ---: | ---: |
113
+ | NOT_RELEVANT | 787 | 32 |
114
+ | RELEVANT | 78 | 157 |
115
+
116
+ ### embedding-svm_sentence_embeddings at threshold 0.500
117
+
118
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
119
+ | --- | ---: | ---: |
120
+ | NOT_RELEVANT | 802 | 17 |
121
+ | RELEVANT | 95 | 140 |
122
+
123
+ ### embedding-svm_sentence_embeddings at threshold 0.401
124
+
125
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
126
+ | --- | ---: | ---: |
127
+ | NOT_RELEVANT | 791 | 28 |
128
+ | RELEVANT | 75 | 160 |
129
+
130
+ ### embedding-lightgbm_sentence_embeddings at threshold 0.500
131
+
132
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
133
+ | --- | ---: | ---: |
134
+ | NOT_RELEVANT | 795 | 24 |
135
+ | RELEVANT | 74 | 161 |
136
+
137
+ ### embedding-lightgbm_sentence_embeddings at threshold 0.145
138
+
139
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
140
+ | --- | ---: | ---: |
141
+ | NOT_RELEVANT | 769 | 50 |
142
+ | RELEVANT | 54 | 181 |
143
+
144
+ ### transformer at threshold 0.500
145
+
146
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
147
+ | --- | ---: | ---: |
148
+ | NOT_RELEVANT | 796 | 23 |
149
+ | RELEVANT | 46 | 189 |
150
+
151
+ ### transformer at threshold 0.328
152
+
153
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
154
+ | --- | ---: | ---: |
155
+ | NOT_RELEVANT | 793 | 26 |
156
+ | RELEVANT | 43 | 192 |
157
+
158
+
159
+ ## Validation-Tuned Thresholds
160
+
161
+ - `logistic_tfidf`: threshold `0.586` (validation F1 `0.798`); test F1 change vs 0.5: `-0.007`.
162
+ - `xgboost_tfidf`: threshold `0.379` (validation F1 `0.898`); test F1 change vs 0.5: `+0.018`.
163
+ - `embedding-logistic_sentence_embeddings`: threshold `0.744` (validation F1 `0.812`); test F1 change vs 0.5: `-0.021`.
164
+ - `embedding-svm_sentence_embeddings`: threshold `0.401` (validation F1 `0.791`); test F1 change vs 0.5: `+0.042`.
165
+ - `embedding-lightgbm_sentence_embeddings`: threshold `0.145` (validation F1 `0.813`); test F1 change vs 0.5: `+0.010`.
166
+ - `transformer`: threshold `0.328` (validation F1 `0.900`); test F1 change vs 0.5: `+0.002`.
167
+
168
+ ## Artifacts
169
+
170
+ - `logistic_tfidf`: `/content/agri-maize_corn-classifier/baselines/logistic`
171
+ - `xgboost_tfidf`: `/content/agri-maize_corn-classifier/baselines/xgboost`
172
+ - `embedding-logistic_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-logistic`
173
+ - `embedding-svm_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-svm`
174
+ - `embedding-lightgbm_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-lightgbm`
175
+ - `transformer`: `/content/agri-maize_corn-classifier/transformer`
176
+
177
+ ## Inference
178
+
179
+ Install the runtime dependencies:
180
+
181
+ ```bash
182
+ pip install transformers torch huggingface_hub pandas joblib scikit-learn xgboost lightgbm
183
+ ```
184
+
185
+ ### Transformer
186
+
187
+ ```python
188
+ import torch
189
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
190
+
191
+ MODEL_ID = "faodl/agri-maize_corn-classifier"
192
+
193
+ texts = [
194
+ "Rice export prices increased after new procurement rules were announced.",
195
+ "The finance ministry released its monthly fuel tax bulletin.",
196
+ ]
197
+
198
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, subfolder="transformer")
199
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, subfolder="transformer")
200
+ threshold = float(getattr(model.config, "threshold", 0.5))
201
+
202
+ encoded = tokenizer(
203
+ texts,
204
+ truncation=True,
205
+ padding=True,
206
+ max_length=256,
207
+ return_tensors="pt",
208
+ )
209
+
210
+ with torch.no_grad():
211
+ logits = model(**encoded).logits
212
+ probabilities = torch.softmax(logits, dim=-1)[:, 1].tolist()
213
+
214
+ for text, probability in zip(texts, probabilities):
215
+ label = model.config.id2label[int(probability >= threshold)]
216
+ print({"text": text, "probability_positive": probability, "label": label})
217
+ ```
218
+
219
+ ### TF-IDF Baselines
220
+
221
+ Available baseline names in this run: "logistic", "xgboost".
222
+
223
+ ```python
224
+ import json
225
+ import joblib
226
+ from huggingface_hub import hf_hub_download
227
+
228
+ MODEL_ID = "faodl/agri-maize_corn-classifier"
229
+ BASELINE = "logistic"
230
+
231
+ texts = [
232
+ "Maize production forecasts were revised after delayed rains.",
233
+ "The central bank published new exchange rate statistics.",
234
+ ]
235
+
236
+ model_path = hf_hub_download(
237
+ repo_id=MODEL_ID,
238
+ repo_type="model",
239
+ filename=f"baselines/{BASELINE}/{BASELINE}_tfidf.joblib",
240
+ )
241
+ report_path = hf_hub_download(
242
+ repo_id=MODEL_ID,
243
+ repo_type="model",
244
+ filename="report.json",
245
+ )
246
+
247
+ pipeline = joblib.load(model_path)
248
+ with open(report_path, encoding="utf-8") as handle:
249
+ report = json.load(handle)
250
+
251
+ threshold = next(
252
+ result["validation_best_threshold"]["threshold"]
253
+ for result in report["results"]
254
+ if result["model_type"] == f"{BASELINE}_tfidf"
255
+ )
256
+
257
+ probabilities = pipeline.predict_proba(texts)[:, 1]
258
+ for text, probability in zip(texts, probabilities):
259
+ label = "RELEVANT" if probability >= threshold else "NOT_RELEVANT"
260
+ print({"text": text, "probability_positive": float(probability), "label": label})
261
+ ```
262
+
263
+ ### Sentence-Embedding Baselines
264
+
265
+ Available embedding baseline names in this run: "embedding-logistic", "embedding-svm", "embedding-lightgbm".
266
+
267
+ ```python
268
+ import joblib
269
+ import torch
270
+ from huggingface_hub import hf_hub_download
271
+ from transformers import AutoModel, AutoTokenizer
272
+
273
+ MODEL_ID = "faodl/agri-maize_corn-classifier"
274
+ BASELINE = "embedding-logistic"
275
+
276
+ texts = [
277
+ "Wheat export inspections rose as demand from importers increased.",
278
+ "The sports ministry announced a new stadium renovation plan.",
279
+ ]
280
+
281
+ model_path = hf_hub_download(
282
+ repo_id=MODEL_ID,
283
+ repo_type="model",
284
+ filename=f"baselines/{BASELINE}/{BASELINE}.joblib",
285
+ )
286
+ artifact = joblib.load(model_path)
287
+ tokenizer = AutoTokenizer.from_pretrained(artifact["embedding_model_name"])
288
+ encoder = AutoModel.from_pretrained(artifact["embedding_model_name"])
289
+ encoder.eval()
290
+
291
+ encoded_batches = []
292
+ batch_size = artifact.get("embedding_batch_size", 64)
293
+ for start in range(0, len(texts), batch_size):
294
+ batch_texts = texts[start : start + batch_size]
295
+ inputs = tokenizer(
296
+ batch_texts,
297
+ padding=True,
298
+ truncation=True,
299
+ max_length=artifact.get("embedding_max_length", 256),
300
+ return_tensors="pt",
301
+ )
302
+ with torch.no_grad():
303
+ outputs = encoder(**inputs)
304
+ token_embeddings = outputs.last_hidden_state
305
+ attention_mask = inputs["attention_mask"].unsqueeze(-1).to(token_embeddings.dtype)
306
+ embeddings = (token_embeddings * attention_mask).sum(dim=1)
307
+ embeddings = embeddings / attention_mask.sum(dim=1).clamp(min=1e-9)
308
+ if artifact.get("normalize_embeddings", True):
309
+ embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=1)
310
+ encoded_batches.append(embeddings)
311
+ embeddings = torch.cat(encoded_batches).numpy()
312
+ probabilities = artifact["classifier"].predict_proba(embeddings)[:, 1]
313
+ threshold = artifact["validation_best_threshold"]["threshold"]
314
+
315
+ for text, probability in zip(texts, probabilities):
316
+ label = "RELEVANT" if probability >= threshold else "NOT_RELEVANT"
317
+ print({"text": text, "probability_positive": float(probability), "label": label})
318
+ ```
319
+
320
+ ## Files
321
+
322
+ - `REPORT.md`: Markdown report for this training run.
323
+ - `report.json`: Machine-readable report containing metrics and thresholds.
324
+ - `transformer/`: Fine-tuned Transformer artifacts, when Transformer training is enabled.
325
+ - `baselines/`: TF-IDF and sentence-embedding baseline artifacts, when baseline training is enabled.
326
+ - `*/validation_predictions.csv` and `*/test_predictions.csv`: Split-level predictions.
REPORT.md ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # AMIS Commodity Classifier Training Report
2
+
3
+ - Dataset: `faodl/amis-agri-maize_corn`
4
+ - Dataset subset: ``
5
+ - Dataset revision: `main`
6
+ - Text column: `chunk_text`
7
+ - Label column: `label`
8
+ - Transformer: `FacebookAI/xlm-roberta-base`
9
+ - Generated at: `2026-06-05T20:32:51.106165+00:00`
10
+
11
+ ## Dataset Summary
12
+
13
+ | Split | Rows | Label 0 | Label 1 | Unique groups | Mean text length |
14
+ | --- | ---: | ---: | ---: | ---: | ---: |
15
+ | train | 4724 | 3822 | 902 | 2226 | 702.9 |
16
+ | validation | 1060 | 843 | 217 | 477 | 708.3 |
17
+ | test | 1054 | 819 | 235 | 478 | 711.9 |
18
+
19
+ ## Threshold Comparison on Validation Split
20
+
21
+ Validation metrics document threshold selection and tuning behavior; test metrics remain the primary estimate of out-of-sample performance.
22
+
23
+ | Model | Threshold | Accuracy | Precision | Recall | F1 | ROC AUC | Average precision |
24
+ | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
25
+ | logistic_tfidf | 0.500 | 0.896 | 0.700 | 0.862 | 0.773 | 0.929 | 0.841 |
26
+ | logistic_tfidf | 0.586 | 0.915 | 0.777 | 0.820 | 0.798 | 0.929 | 0.841 |
27
+ | xgboost_tfidf | 0.500 | 0.957 | 0.913 | 0.871 | 0.892 | 0.967 | 0.914 |
28
+ | xgboost_tfidf | 0.379 | 0.958 | 0.902 | 0.894 | 0.898 | 0.967 | 0.914 |
29
+ | embedding-logistic_sentence_embeddings | 0.500 | 0.881 | 0.649 | 0.912 | 0.759 | 0.959 | 0.867 |
30
+ | embedding-logistic_sentence_embeddings | 0.744 | 0.923 | 0.808 | 0.816 | 0.812 | 0.959 | 0.867 |
31
+ | embedding-svm_sentence_embeddings | 0.500 | 0.913 | 0.849 | 0.700 | 0.768 | 0.955 | 0.858 |
32
+ | embedding-svm_sentence_embeddings | 0.401 | 0.914 | 0.789 | 0.793 | 0.791 | 0.955 | 0.858 |
33
+ | embedding-lightgbm_sentence_embeddings | 0.500 | 0.916 | 0.791 | 0.802 | 0.796 | 0.963 | 0.878 |
34
+ | embedding-lightgbm_sentence_embeddings | 0.145 | 0.916 | 0.746 | 0.894 | 0.813 | 0.963 | 0.878 |
35
+ | transformer | 0.500 | 0.958 | 0.913 | 0.876 | 0.894 | 0.973 | 0.943 |
36
+ | transformer | 0.328 | 0.959 | 0.907 | 0.894 | 0.900 | 0.973 | 0.943 |
37
+
38
+ ## Threshold Comparison on Test Split
39
+
40
+ | Model | Threshold | Accuracy | Precision | Recall | F1 | ROC AUC | Average precision |
41
+ | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
42
+ | logistic_tfidf | 0.500 | 0.910 | 0.787 | 0.817 | 0.802 | 0.953 | 0.863 |
43
+ | logistic_tfidf | 0.586 | 0.915 | 0.857 | 0.740 | 0.795 | 0.953 | 0.863 |
44
+ | xgboost_tfidf | 0.500 | 0.942 | 0.914 | 0.817 | 0.863 | 0.968 | 0.920 |
45
+ | xgboost_tfidf | 0.379 | 0.948 | 0.895 | 0.868 | 0.881 | 0.968 | 0.920 |
46
+ | embedding-logistic_sentence_embeddings | 0.500 | 0.884 | 0.704 | 0.830 | 0.762 | 0.936 | 0.844 |
47
+ | embedding-logistic_sentence_embeddings | 0.744 | 0.896 | 0.831 | 0.668 | 0.741 | 0.936 | 0.844 |
48
+ | embedding-svm_sentence_embeddings | 0.500 | 0.894 | 0.892 | 0.596 | 0.714 | 0.932 | 0.842 |
49
+ | embedding-svm_sentence_embeddings | 0.401 | 0.902 | 0.851 | 0.681 | 0.757 | 0.932 | 0.842 |
50
+ | embedding-lightgbm_sentence_embeddings | 0.500 | 0.907 | 0.870 | 0.685 | 0.767 | 0.953 | 0.873 |
51
+ | embedding-lightgbm_sentence_embeddings | 0.145 | 0.901 | 0.784 | 0.770 | 0.777 | 0.953 | 0.873 |
52
+ | transformer | 0.500 | 0.935 | 0.892 | 0.804 | 0.846 | 0.953 | 0.890 |
53
+ | transformer | 0.328 | 0.935 | 0.881 | 0.817 | 0.848 | 0.953 | 0.890 |
54
+
55
+ ## Confusion Matrices on Test Split
56
+
57
+ Rows are true labels and columns are predicted labels.
58
+
59
+ ### logistic_tfidf at threshold 0.500
60
+
61
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
62
+ | --- | ---: | ---: |
63
+ | NOT_RELEVANT | 767 | 52 |
64
+ | RELEVANT | 43 | 192 |
65
+
66
+ ### logistic_tfidf at threshold 0.586
67
+
68
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
69
+ | --- | ---: | ---: |
70
+ | NOT_RELEVANT | 790 | 29 |
71
+ | RELEVANT | 61 | 174 |
72
+
73
+ ### xgboost_tfidf at threshold 0.500
74
+
75
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
76
+ | --- | ---: | ---: |
77
+ | NOT_RELEVANT | 801 | 18 |
78
+ | RELEVANT | 43 | 192 |
79
+
80
+ ### xgboost_tfidf at threshold 0.379
81
+
82
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
83
+ | --- | ---: | ---: |
84
+ | NOT_RELEVANT | 795 | 24 |
85
+ | RELEVANT | 31 | 204 |
86
+
87
+ ### embedding-logistic_sentence_embeddings at threshold 0.500
88
+
89
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
90
+ | --- | ---: | ---: |
91
+ | NOT_RELEVANT | 737 | 82 |
92
+ | RELEVANT | 40 | 195 |
93
+
94
+ ### embedding-logistic_sentence_embeddings at threshold 0.744
95
+
96
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
97
+ | --- | ---: | ---: |
98
+ | NOT_RELEVANT | 787 | 32 |
99
+ | RELEVANT | 78 | 157 |
100
+
101
+ ### embedding-svm_sentence_embeddings at threshold 0.500
102
+
103
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
104
+ | --- | ---: | ---: |
105
+ | NOT_RELEVANT | 802 | 17 |
106
+ | RELEVANT | 95 | 140 |
107
+
108
+ ### embedding-svm_sentence_embeddings at threshold 0.401
109
+
110
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
111
+ | --- | ---: | ---: |
112
+ | NOT_RELEVANT | 791 | 28 |
113
+ | RELEVANT | 75 | 160 |
114
+
115
+ ### embedding-lightgbm_sentence_embeddings at threshold 0.500
116
+
117
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
118
+ | --- | ---: | ---: |
119
+ | NOT_RELEVANT | 795 | 24 |
120
+ | RELEVANT | 74 | 161 |
121
+
122
+ ### embedding-lightgbm_sentence_embeddings at threshold 0.145
123
+
124
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
125
+ | --- | ---: | ---: |
126
+ | NOT_RELEVANT | 769 | 50 |
127
+ | RELEVANT | 54 | 181 |
128
+
129
+ ### transformer at threshold 0.500
130
+
131
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
132
+ | --- | ---: | ---: |
133
+ | NOT_RELEVANT | 796 | 23 |
134
+ | RELEVANT | 46 | 189 |
135
+
136
+ ### transformer at threshold 0.328
137
+
138
+ | True / Predicted | NOT_RELEVANT | RELEVANT |
139
+ | --- | ---: | ---: |
140
+ | NOT_RELEVANT | 793 | 26 |
141
+ | RELEVANT | 43 | 192 |
142
+
143
+
144
+ ## Validation-Tuned Thresholds
145
+
146
+ - `logistic_tfidf`: threshold `0.586` (validation F1 `0.798`); test F1 change vs 0.5: `-0.007`.
147
+ - `xgboost_tfidf`: threshold `0.379` (validation F1 `0.898`); test F1 change vs 0.5: `+0.018`.
148
+ - `embedding-logistic_sentence_embeddings`: threshold `0.744` (validation F1 `0.812`); test F1 change vs 0.5: `-0.021`.
149
+ - `embedding-svm_sentence_embeddings`: threshold `0.401` (validation F1 `0.791`); test F1 change vs 0.5: `+0.042`.
150
+ - `embedding-lightgbm_sentence_embeddings`: threshold `0.145` (validation F1 `0.813`); test F1 change vs 0.5: `+0.010`.
151
+ - `transformer`: threshold `0.328` (validation F1 `0.900`); test F1 change vs 0.5: `+0.002`.
152
+
153
+ ## Artifacts
154
+
155
+ - `logistic_tfidf`: `/content/agri-maize_corn-classifier/baselines/logistic`
156
+ - `xgboost_tfidf`: `/content/agri-maize_corn-classifier/baselines/xgboost`
157
+ - `embedding-logistic_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-logistic`
158
+ - `embedding-svm_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-svm`
159
+ - `embedding-lightgbm_sentence_embeddings`: `/content/agri-maize_corn-classifier/baselines/embedding-lightgbm`
160
+ - `transformer`: `/content/agri-maize_corn-classifier/transformer`
baselines/embedding-lightgbm/embedding-lightgbm.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b43a5ce8df629f94792b83a7b1b2eed85e91ff4150a84a444e907ff8363eba3f
3
+ size 1457950
baselines/embedding-lightgbm/test_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/embedding-lightgbm/validation_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/embedding-logistic/embedding-logistic.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81ed45a29852cc67656a4e024605be056bd1408302937b32131c73078e3fef7e
3
+ size 2821
baselines/embedding-logistic/test_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/embedding-logistic/validation_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/embedding-svm/embedding-svm.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2e256f5b8bc28bf4f9abc3cd3c02c2a9cf7605f86c444d37761bd0a4b2c6c30
3
+ size 11770
baselines/embedding-svm/test_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/embedding-svm/validation_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/logistic/logistic_tfidf.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e2235e9e06a71e7da44c278b563f426835cfa451f79405cfca9da16b3b056a0
3
+ size 2453428
baselines/logistic/test_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/logistic/validation_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/xgboost/test_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/xgboost/validation_predictions.csv ADDED
The diff for this file is too large to render. See raw diff
 
baselines/xgboost/xgboost_tfidf.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:442a7c91a42548dc025003eb705c5a861953a22dc419ecbd960a7f4506e25f6c
3
+ size 2540068
report.json ADDED
@@ -0,0 +1,1257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "created_at": "2026-06-05T20:32:51.106165+00:00",
3
+ "config": {
4
+ "hf_dataset": "faodl/amis-agri-maize_corn",
5
+ "hf_subset": null,
6
+ "hf_revision": "main",
7
+ "train_split": "train",
8
+ "validation_split": "validation",
9
+ "test_split": "test",
10
+ "text_col": "chunk_text",
11
+ "label_col": "label",
12
+ "group_col": "id",
13
+ "id_col": "chunk_id",
14
+ "model_name": "FacebookAI/xlm-roberta-base",
15
+ "output_dir": "/content/agri-maize_corn-classifier",
16
+ "max_length": 256,
17
+ "learning_rate": 2e-05,
18
+ "weight_decay": 0.01,
19
+ "num_train_epochs": 5.0,
20
+ "per_device_train_batch_size": 16,
21
+ "per_device_eval_batch_size": 32,
22
+ "gradient_accumulation_steps": 1,
23
+ "warmup_ratio": 0.1,
24
+ "early_stopping_patience": 2,
25
+ "seed": 42,
26
+ "metric_for_best_model": "f1",
27
+ "skip_transformer": false,
28
+ "skip_baselines": false,
29
+ "baseline_models": [
30
+ "logistic",
31
+ "xgboost",
32
+ "embedding-logistic",
33
+ "embedding-svm",
34
+ "embedding-lightgbm"
35
+ ],
36
+ "tfidf_max_features": 50000,
37
+ "embedding_model_name": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
38
+ "embedding_batch_size": 64,
39
+ "positive_label_name": "RELEVANT",
40
+ "negative_label_name": "NOT_RELEVANT",
41
+ "push_to_hub": true,
42
+ "hub_model_id": "faodl/agri-maize_corn-classifier",
43
+ "hub_private_repo": false
44
+ },
45
+ "dataset_summary": {
46
+ "train": {
47
+ "rows": 4724,
48
+ "labels": {
49
+ "0": 3822,
50
+ "1": 902
51
+ },
52
+ "unique_groups": 2226,
53
+ "text_length_mean": 702.8679085520745,
54
+ "text_length_median": 794.0
55
+ },
56
+ "validation": {
57
+ "rows": 1060,
58
+ "labels": {
59
+ "0": 843,
60
+ "1": 217
61
+ },
62
+ "unique_groups": 477,
63
+ "text_length_mean": 708.2584905660377,
64
+ "text_length_median": 795.0
65
+ },
66
+ "test": {
67
+ "rows": 1054,
68
+ "labels": {
69
+ "0": 819,
70
+ "1": 235
71
+ },
72
+ "unique_groups": 478,
73
+ "text_length_mean": 711.8595825426945,
74
+ "text_length_median": 794.0
75
+ }
76
+ },
77
+ "results": [
78
+ {
79
+ "model_type": "logistic_tfidf",
80
+ "model_name": "logistic",
81
+ "artifact_dir": "/content/agri-maize_corn-classifier/baselines/logistic",
82
+ "artifact_file": "/content/agri-maize_corn-classifier/baselines/logistic/logistic_tfidf.joblib",
83
+ "validation_best_threshold": {
84
+ "threshold": 0.5864890967583224,
85
+ "f1": 0.7982062780269059,
86
+ "precision": 0.777292576419214,
87
+ "recall": 0.8202764976958525
88
+ },
89
+ "validation_default_0_5": {
90
+ "threshold": 0.5,
91
+ "accuracy": 0.8962264150943396,
92
+ "precision": 0.700374531835206,
93
+ "recall": 0.8617511520737328,
94
+ "f1": 0.7727272727272727,
95
+ "confusion_matrix": [
96
+ [
97
+ 763,
98
+ 80
99
+ ],
100
+ [
101
+ 30,
102
+ 187
103
+ ]
104
+ ],
105
+ "classification_report": {
106
+ "NOT_RELEVANT": {
107
+ "precision": 0.9621689785624212,
108
+ "recall": 0.9051008303677343,
109
+ "f1-score": 0.9327628361858191,
110
+ "support": 843.0
111
+ },
112
+ "RELEVANT": {
113
+ "precision": 0.700374531835206,
114
+ "recall": 0.8617511520737328,
115
+ "f1-score": 0.7727272727272727,
116
+ "support": 217.0
117
+ },
118
+ "accuracy": 0.8962264150943396,
119
+ "macro avg": {
120
+ "precision": 0.8312717551988136,
121
+ "recall": 0.8834259912207335,
122
+ "f1-score": 0.8527450544565459,
123
+ "support": 1060.0
124
+ },
125
+ "weighted avg": {
126
+ "precision": 0.9085752097512838,
127
+ "recall": 0.8962264150943396,
128
+ "f1-score": 0.9000008387608148,
129
+ "support": 1060.0
130
+ }
131
+ },
132
+ "roc_auc": 0.9288420224018894,
133
+ "average_precision": 0.8406024632295316
134
+ },
135
+ "validation_optimal_threshold": {
136
+ "threshold": 0.5864890967583224,
137
+ "accuracy": 0.9150943396226415,
138
+ "precision": 0.777292576419214,
139
+ "recall": 0.8202764976958525,
140
+ "f1": 0.7982062780269058,
141
+ "confusion_matrix": [
142
+ [
143
+ 792,
144
+ 51
145
+ ],
146
+ [
147
+ 39,
148
+ 178
149
+ ]
150
+ ],
151
+ "classification_report": {
152
+ "NOT_RELEVANT": {
153
+ "precision": 0.9530685920577617,
154
+ "recall": 0.9395017793594306,
155
+ "f1-score": 0.946236559139785,
156
+ "support": 843.0
157
+ },
158
+ "RELEVANT": {
159
+ "precision": 0.777292576419214,
160
+ "recall": 0.8202764976958525,
161
+ "f1-score": 0.7982062780269058,
162
+ "support": 217.0
163
+ },
164
+ "accuracy": 0.9150943396226415,
165
+ "macro avg": {
166
+ "precision": 0.8651805842384879,
167
+ "recall": 0.8798891385276415,
168
+ "f1-score": 0.8722214185833455,
169
+ "support": 1060.0
170
+ },
171
+ "weighted avg": {
172
+ "precision": 0.9170842567808138,
173
+ "recall": 0.9150943396226415,
174
+ "f1-score": 0.9159322468742239,
175
+ "support": 1060.0
176
+ }
177
+ },
178
+ "roc_auc": 0.9288420224018894,
179
+ "average_precision": 0.8406024632295316
180
+ },
181
+ "test_default_0_5": {
182
+ "threshold": 0.5,
183
+ "accuracy": 0.9098671726755219,
184
+ "precision": 0.7868852459016393,
185
+ "recall": 0.8170212765957446,
186
+ "f1": 0.8016701461377871,
187
+ "confusion_matrix": [
188
+ [
189
+ 767,
190
+ 52
191
+ ],
192
+ [
193
+ 43,
194
+ 192
195
+ ]
196
+ ],
197
+ "classification_report": {
198
+ "NOT_RELEVANT": {
199
+ "precision": 0.9469135802469136,
200
+ "recall": 0.9365079365079365,
201
+ "f1-score": 0.941682013505218,
202
+ "support": 819.0
203
+ },
204
+ "RELEVANT": {
205
+ "precision": 0.7868852459016393,
206
+ "recall": 0.8170212765957446,
207
+ "f1-score": 0.8016701461377871,
208
+ "support": 235.0
209
+ },
210
+ "accuracy": 0.9098671726755219,
211
+ "macro avg": {
212
+ "precision": 0.8668994130742764,
213
+ "recall": 0.8767646065518406,
214
+ "f1-score": 0.8716760798215025,
215
+ "support": 1054.0
216
+ },
217
+ "weighted avg": {
218
+ "precision": 0.9112336385285649,
219
+ "recall": 0.9098671726755219,
220
+ "f1-score": 0.9104649463028021,
221
+ "support": 1054.0
222
+ }
223
+ },
224
+ "roc_auc": 0.9531551191125659,
225
+ "average_precision": 0.8625249758747111
226
+ },
227
+ "test_optimal_threshold": {
228
+ "threshold": 0.5864890967583224,
229
+ "accuracy": 0.9146110056925996,
230
+ "precision": 0.8571428571428571,
231
+ "recall": 0.7404255319148936,
232
+ "f1": 0.7945205479452054,
233
+ "confusion_matrix": [
234
+ [
235
+ 790,
236
+ 29
237
+ ],
238
+ [
239
+ 61,
240
+ 174
241
+ ]
242
+ ],
243
+ "classification_report": {
244
+ "NOT_RELEVANT": {
245
+ "precision": 0.9283196239717979,
246
+ "recall": 0.9645909645909646,
247
+ "f1-score": 0.9461077844311377,
248
+ "support": 819.0
249
+ },
250
+ "RELEVANT": {
251
+ "precision": 0.8571428571428571,
252
+ "recall": 0.7404255319148936,
253
+ "f1-score": 0.7945205479452054,
254
+ "support": 235.0
255
+ },
256
+ "accuracy": 0.9146110056925996,
257
+ "macro avg": {
258
+ "precision": 0.8927312405573276,
259
+ "recall": 0.8525082482529291,
260
+ "f1-score": 0.8703141661881716,
261
+ "support": 1054.0
262
+ },
263
+ "weighted avg": {
264
+ "precision": 0.912450041234795,
265
+ "recall": 0.9146110056925996,
266
+ "f1-score": 0.9123098711728891,
267
+ "support": 1054.0
268
+ }
269
+ },
270
+ "roc_auc": 0.9531551191125659,
271
+ "average_precision": 0.8625249758747111
272
+ }
273
+ },
274
+ {
275
+ "model_type": "xgboost_tfidf",
276
+ "model_name": "xgboost",
277
+ "artifact_dir": "/content/agri-maize_corn-classifier/baselines/xgboost",
278
+ "artifact_file": "/content/agri-maize_corn-classifier/baselines/xgboost/xgboost_tfidf.joblib",
279
+ "validation_best_threshold": {
280
+ "threshold": 0.37899935245513916,
281
+ "f1": 0.8981481481481481,
282
+ "precision": 0.9023255813953488,
283
+ "recall": 0.8940092165898618
284
+ },
285
+ "validation_default_0_5": {
286
+ "threshold": 0.5,
287
+ "accuracy": 0.9566037735849057,
288
+ "precision": 0.9130434782608695,
289
+ "recall": 0.8709677419354839,
290
+ "f1": 0.8915094339622641,
291
+ "confusion_matrix": [
292
+ [
293
+ 825,
294
+ 18
295
+ ],
296
+ [
297
+ 28,
298
+ 189
299
+ ]
300
+ ],
301
+ "classification_report": {
302
+ "NOT_RELEVANT": {
303
+ "precision": 0.9671746776084408,
304
+ "recall": 0.9786476868327402,
305
+ "f1-score": 0.972877358490566,
306
+ "support": 843.0
307
+ },
308
+ "RELEVANT": {
309
+ "precision": 0.9130434782608695,
310
+ "recall": 0.8709677419354839,
311
+ "f1-score": 0.8915094339622641,
312
+ "support": 217.0
313
+ },
314
+ "accuracy": 0.9566037735849057,
315
+ "macro avg": {
316
+ "precision": 0.9401090779346551,
317
+ "recall": 0.924807714384112,
318
+ "f1-score": 0.9321933962264151,
319
+ "support": 1060.0
320
+ },
321
+ "weighted avg": {
322
+ "precision": 0.9560931018929474,
323
+ "recall": 0.9566037735849057,
324
+ "f1-score": 0.9562199626201495,
325
+ "support": 1060.0
326
+ }
327
+ },
328
+ "roc_auc": 0.966850889133061,
329
+ "average_precision": 0.9141254109659447
330
+ },
331
+ "validation_optimal_threshold": {
332
+ "threshold": 0.37899935245513916,
333
+ "accuracy": 0.9584905660377359,
334
+ "precision": 0.9023255813953488,
335
+ "recall": 0.8940092165898618,
336
+ "f1": 0.8981481481481481,
337
+ "confusion_matrix": [
338
+ [
339
+ 822,
340
+ 21
341
+ ],
342
+ [
343
+ 23,
344
+ 194
345
+ ]
346
+ ],
347
+ "classification_report": {
348
+ "NOT_RELEVANT": {
349
+ "precision": 0.9727810650887574,
350
+ "recall": 0.9750889679715302,
351
+ "f1-score": 0.9739336492890995,
352
+ "support": 843.0
353
+ },
354
+ "RELEVANT": {
355
+ "precision": 0.9023255813953488,
356
+ "recall": 0.8940092165898618,
357
+ "f1-score": 0.8981481481481481,
358
+ "support": 217.0
359
+ },
360
+ "accuracy": 0.9584905660377359,
361
+ "macro avg": {
362
+ "precision": 0.9375533232420531,
363
+ "recall": 0.934549092280696,
364
+ "f1-score": 0.9360408987186238,
365
+ "support": 1060.0
366
+ },
367
+ "weighted avg": {
368
+ "precision": 0.9583576311628426,
369
+ "recall": 0.9584905660377359,
370
+ "f1-score": 0.9584190702819425,
371
+ "support": 1060.0
372
+ }
373
+ },
374
+ "roc_auc": 0.966850889133061,
375
+ "average_precision": 0.9141254109659447
376
+ },
377
+ "test_default_0_5": {
378
+ "threshold": 0.5,
379
+ "accuracy": 0.9421252371916509,
380
+ "precision": 0.9142857142857143,
381
+ "recall": 0.8170212765957446,
382
+ "f1": 0.8629213483146068,
383
+ "confusion_matrix": [
384
+ [
385
+ 801,
386
+ 18
387
+ ],
388
+ [
389
+ 43,
390
+ 192
391
+ ]
392
+ ],
393
+ "classification_report": {
394
+ "NOT_RELEVANT": {
395
+ "precision": 0.9490521327014217,
396
+ "recall": 0.978021978021978,
397
+ "f1-score": 0.963319302465424,
398
+ "support": 819.0
399
+ },
400
+ "RELEVANT": {
401
+ "precision": 0.9142857142857143,
402
+ "recall": 0.8170212765957446,
403
+ "f1-score": 0.8629213483146068,
404
+ "support": 235.0
405
+ },
406
+ "accuracy": 0.9421252371916509,
407
+ "macro avg": {
408
+ "precision": 0.931668923493568,
409
+ "recall": 0.8975216273088613,
410
+ "f1-score": 0.9131203253900153,
411
+ "support": 1054.0
412
+ },
413
+ "weighted avg": {
414
+ "precision": 0.941300606773821,
415
+ "recall": 0.9421252371916509,
416
+ "f1-score": 0.9409345593672817,
417
+ "support": 1054.0
418
+ }
419
+ },
420
+ "roc_auc": 0.9682098043800171,
421
+ "average_precision": 0.9201070133275652
422
+ },
423
+ "test_optimal_threshold": {
424
+ "threshold": 0.37899935245513916,
425
+ "accuracy": 0.9478178368121443,
426
+ "precision": 0.8947368421052632,
427
+ "recall": 0.8680851063829788,
428
+ "f1": 0.8812095032397408,
429
+ "confusion_matrix": [
430
+ [
431
+ 795,
432
+ 24
433
+ ],
434
+ [
435
+ 31,
436
+ 204
437
+ ]
438
+ ],
439
+ "classification_report": {
440
+ "NOT_RELEVANT": {
441
+ "precision": 0.9624697336561744,
442
+ "recall": 0.9706959706959707,
443
+ "f1-score": 0.9665653495440729,
444
+ "support": 819.0
445
+ },
446
+ "RELEVANT": {
447
+ "precision": 0.8947368421052632,
448
+ "recall": 0.8680851063829788,
449
+ "f1-score": 0.8812095032397408,
450
+ "support": 235.0
451
+ },
452
+ "accuracy": 0.9478178368121443,
453
+ "macro avg": {
454
+ "precision": 0.9286032878807188,
455
+ "recall": 0.9193905385394747,
456
+ "f1-score": 0.9238874263919068,
457
+ "support": 1054.0
458
+ },
459
+ "weighted avg": {
460
+ "precision": 0.9473679978739503,
461
+ "recall": 0.9478178368121443,
462
+ "f1-score": 0.9475343970948148,
463
+ "support": 1054.0
464
+ }
465
+ },
466
+ "roc_auc": 0.9682098043800171,
467
+ "average_precision": 0.9201070133275652
468
+ }
469
+ },
470
+ {
471
+ "model_type": "embedding-logistic_sentence_embeddings",
472
+ "model_name": "logistic",
473
+ "embedding_model_name": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
474
+ "artifact_dir": "/content/agri-maize_corn-classifier/baselines/embedding-logistic",
475
+ "artifact_file": "/content/agri-maize_corn-classifier/baselines/embedding-logistic/embedding-logistic.joblib",
476
+ "validation_best_threshold": {
477
+ "threshold": 0.7437081336975098,
478
+ "f1": 0.8119266055045872,
479
+ "precision": 0.8082191780821918,
480
+ "recall": 0.815668202764977
481
+ },
482
+ "validation_default_0_5": {
483
+ "threshold": 0.5,
484
+ "accuracy": 0.8811320754716981,
485
+ "precision": 0.6491803278688525,
486
+ "recall": 0.9124423963133641,
487
+ "f1": 0.7586206896551724,
488
+ "confusion_matrix": [
489
+ [
490
+ 736,
491
+ 107
492
+ ],
493
+ [
494
+ 19,
495
+ 198
496
+ ]
497
+ ],
498
+ "classification_report": {
499
+ "NOT_RELEVANT": {
500
+ "precision": 0.9748344370860927,
501
+ "recall": 0.8730723606168446,
502
+ "f1-score": 0.9211514392991239,
503
+ "support": 843.0
504
+ },
505
+ "RELEVANT": {
506
+ "precision": 0.6491803278688525,
507
+ "recall": 0.9124423963133641,
508
+ "f1-score": 0.7586206896551724,
509
+ "support": 217.0
510
+ },
511
+ "accuracy": 0.8811320754716981,
512
+ "macro avg": {
513
+ "precision": 0.8120073824774726,
514
+ "recall": 0.8927573784651044,
515
+ "f1-score": 0.8398860644771482,
516
+ "support": 1060.0
517
+ },
518
+ "weighted avg": {
519
+ "precision": 0.9081675109538841,
520
+ "recall": 0.8811320754716981,
521
+ "f1-score": 0.887878634890881,
522
+ "support": 1060.0
523
+ }
524
+ },
525
+ "roc_auc": 0.9590938659931886,
526
+ "average_precision": 0.8665525722038933
527
+ },
528
+ "validation_optimal_threshold": {
529
+ "threshold": 0.7437081336975098,
530
+ "accuracy": 0.9226415094339623,
531
+ "precision": 0.8082191780821918,
532
+ "recall": 0.815668202764977,
533
+ "f1": 0.8119266055045872,
534
+ "confusion_matrix": [
535
+ [
536
+ 801,
537
+ 42
538
+ ],
539
+ [
540
+ 40,
541
+ 177
542
+ ]
543
+ ],
544
+ "classification_report": {
545
+ "NOT_RELEVANT": {
546
+ "precision": 0.9524375743162902,
547
+ "recall": 0.9501779359430605,
548
+ "f1-score": 0.9513064133016627,
549
+ "support": 843.0
550
+ },
551
+ "RELEVANT": {
552
+ "precision": 0.8082191780821918,
553
+ "recall": 0.815668202764977,
554
+ "f1-score": 0.8119266055045872,
555
+ "support": 217.0
556
+ },
557
+ "accuracy": 0.9226415094339623,
558
+ "macro avg": {
559
+ "precision": 0.880328376199241,
560
+ "recall": 0.8829230693540188,
561
+ "f1-score": 0.881616509403125,
562
+ "support": 1060.0
563
+ },
564
+ "weighted avg": {
565
+ "precision": 0.9229136196155361,
566
+ "recall": 0.9226415094339623,
567
+ "f1-score": 0.9227729998186764,
568
+ "support": 1060.0
569
+ }
570
+ },
571
+ "roc_auc": 0.9590938659931886,
572
+ "average_precision": 0.8665525722038933
573
+ },
574
+ "test_default_0_5": {
575
+ "threshold": 0.5,
576
+ "accuracy": 0.8842504743833017,
577
+ "precision": 0.703971119133574,
578
+ "recall": 0.8297872340425532,
579
+ "f1": 0.76171875,
580
+ "confusion_matrix": [
581
+ [
582
+ 737,
583
+ 82
584
+ ],
585
+ [
586
+ 40,
587
+ 195
588
+ ]
589
+ ],
590
+ "classification_report": {
591
+ "NOT_RELEVANT": {
592
+ "precision": 0.9485199485199485,
593
+ "recall": 0.8998778998778999,
594
+ "f1-score": 0.9235588972431078,
595
+ "support": 819.0
596
+ },
597
+ "RELEVANT": {
598
+ "precision": 0.703971119133574,
599
+ "recall": 0.8297872340425532,
600
+ "f1-score": 0.76171875,
601
+ "support": 235.0
602
+ },
603
+ "accuracy": 0.8842504743833017,
604
+ "macro avg": {
605
+ "precision": 0.8262455338267612,
606
+ "recall": 0.8648325669602266,
607
+ "f1-score": 0.8426388236215538,
608
+ "support": 1054.0
609
+ },
610
+ "weighted avg": {
611
+ "precision": 0.8939953043968005,
612
+ "recall": 0.8842504743833017,
613
+ "f1-score": 0.8874749934460202,
614
+ "support": 1054.0
615
+ }
616
+ },
617
+ "roc_auc": 0.9363156937625022,
618
+ "average_precision": 0.843737774596255
619
+ },
620
+ "test_optimal_threshold": {
621
+ "threshold": 0.7437081336975098,
622
+ "accuracy": 0.8956356736242884,
623
+ "precision": 0.8306878306878307,
624
+ "recall": 0.6680851063829787,
625
+ "f1": 0.7405660377358491,
626
+ "confusion_matrix": [
627
+ [
628
+ 787,
629
+ 32
630
+ ],
631
+ [
632
+ 78,
633
+ 157
634
+ ]
635
+ ],
636
+ "classification_report": {
637
+ "NOT_RELEVANT": {
638
+ "precision": 0.9098265895953758,
639
+ "recall": 0.960927960927961,
640
+ "f1-score": 0.9346793349168646,
641
+ "support": 819.0
642
+ },
643
+ "RELEVANT": {
644
+ "precision": 0.8306878306878307,
645
+ "recall": 0.6680851063829787,
646
+ "f1-score": 0.7405660377358491,
647
+ "support": 235.0
648
+ },
649
+ "accuracy": 0.8956356736242884,
650
+ "macro avg": {
651
+ "precision": 0.8702572101416033,
652
+ "recall": 0.8145065336554698,
653
+ "f1-score": 0.8376226863263568,
654
+ "support": 1054.0
655
+ },
656
+ "weighted avg": {
657
+ "precision": 0.8921817998958756,
658
+ "recall": 0.8956356736242884,
659
+ "f1-score": 0.8913998047104712,
660
+ "support": 1054.0
661
+ }
662
+ },
663
+ "roc_auc": 0.9363156937625022,
664
+ "average_precision": 0.843737774596255
665
+ }
666
+ },
667
+ {
668
+ "model_type": "embedding-svm_sentence_embeddings",
669
+ "model_name": "svm",
670
+ "embedding_model_name": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
671
+ "artifact_dir": "/content/agri-maize_corn-classifier/baselines/embedding-svm",
672
+ "artifact_file": "/content/agri-maize_corn-classifier/baselines/embedding-svm/embedding-svm.joblib",
673
+ "validation_best_threshold": {
674
+ "threshold": 0.4006770902787147,
675
+ "f1": 0.7908045977011494,
676
+ "precision": 0.7889908256880734,
677
+ "recall": 0.7926267281105991
678
+ },
679
+ "validation_default_0_5": {
680
+ "threshold": 0.5,
681
+ "accuracy": 0.9132075471698113,
682
+ "precision": 0.8491620111731844,
683
+ "recall": 0.7004608294930875,
684
+ "f1": 0.7676767676767676,
685
+ "confusion_matrix": [
686
+ [
687
+ 816,
688
+ 27
689
+ ],
690
+ [
691
+ 65,
692
+ 152
693
+ ]
694
+ ],
695
+ "classification_report": {
696
+ "NOT_RELEVANT": {
697
+ "precision": 0.9262202043132803,
698
+ "recall": 0.9679715302491103,
699
+ "f1-score": 0.9466357308584686,
700
+ "support": 843.0
701
+ },
702
+ "RELEVANT": {
703
+ "precision": 0.8491620111731844,
704
+ "recall": 0.7004608294930875,
705
+ "f1-score": 0.7676767676767676,
706
+ "support": 217.0
707
+ },
708
+ "accuracy": 0.9132075471698113,
709
+ "macro avg": {
710
+ "precision": 0.8876911077432323,
711
+ "recall": 0.834216179871099,
712
+ "f1-score": 0.8571562492676181,
713
+ "support": 1060.0
714
+ },
715
+ "weighted avg": {
716
+ "precision": 0.9104450836421474,
717
+ "recall": 0.9132075471698113,
718
+ "f1-score": 0.9099997921693845,
719
+ "support": 1060.0
720
+ }
721
+ },
722
+ "roc_auc": 0.9551306230217951,
723
+ "average_precision": 0.8583227993086541
724
+ },
725
+ "validation_optimal_threshold": {
726
+ "threshold": 0.4006770902787147,
727
+ "accuracy": 0.9141509433962264,
728
+ "precision": 0.7889908256880734,
729
+ "recall": 0.7926267281105991,
730
+ "f1": 0.7908045977011494,
731
+ "confusion_matrix": [
732
+ [
733
+ 797,
734
+ 46
735
+ ],
736
+ [
737
+ 45,
738
+ 172
739
+ ]
740
+ ],
741
+ "classification_report": {
742
+ "NOT_RELEVANT": {
743
+ "precision": 0.9465558194774347,
744
+ "recall": 0.9454329774614472,
745
+ "f1-score": 0.9459940652818991,
746
+ "support": 843.0
747
+ },
748
+ "RELEVANT": {
749
+ "precision": 0.7889908256880734,
750
+ "recall": 0.7926267281105991,
751
+ "f1-score": 0.7908045977011494,
752
+ "support": 217.0
753
+ },
754
+ "accuracy": 0.9141509433962264,
755
+ "macro avg": {
756
+ "precision": 0.8677733225827541,
757
+ "recall": 0.8690298527860232,
758
+ "f1-score": 0.8683993314915243,
759
+ "support": 1060.0
760
+ },
761
+ "weighted avg": {
762
+ "precision": 0.9142995896167825,
763
+ "recall": 0.9141509433962264,
764
+ "f1-score": 0.914224145975274,
765
+ "support": 1060.0
766
+ }
767
+ },
768
+ "roc_auc": 0.9551306230217951,
769
+ "average_precision": 0.8583227993086541
770
+ },
771
+ "test_default_0_5": {
772
+ "threshold": 0.5,
773
+ "accuracy": 0.8937381404174574,
774
+ "precision": 0.89171974522293,
775
+ "recall": 0.5957446808510638,
776
+ "f1": 0.7142857142857143,
777
+ "confusion_matrix": [
778
+ [
779
+ 802,
780
+ 17
781
+ ],
782
+ [
783
+ 95,
784
+ 140
785
+ ]
786
+ ],
787
+ "classification_report": {
788
+ "NOT_RELEVANT": {
789
+ "precision": 0.8940914158305463,
790
+ "recall": 0.9792429792429792,
791
+ "f1-score": 0.9347319347319347,
792
+ "support": 819.0
793
+ },
794
+ "RELEVANT": {
795
+ "precision": 0.89171974522293,
796
+ "recall": 0.5957446808510638,
797
+ "f1-score": 0.7142857142857143,
798
+ "support": 235.0
799
+ },
800
+ "accuracy": 0.8937381404174574,
801
+ "macro avg": {
802
+ "precision": 0.8929055805267381,
803
+ "recall": 0.7874938300470216,
804
+ "f1-score": 0.8245088245088246,
805
+ "support": 1054.0
806
+ },
807
+ "weighted avg": {
808
+ "precision": 0.8935626277918463,
809
+ "recall": 0.8937381404174574,
810
+ "f1-score": 0.8855812119569235,
811
+ "support": 1054.0
812
+ }
813
+ },
814
+ "roc_auc": 0.9321590938612216,
815
+ "average_precision": 0.8417078389566306
816
+ },
817
+ "test_optimal_threshold": {
818
+ "threshold": 0.4006770902787147,
819
+ "accuracy": 0.9022770398481973,
820
+ "precision": 0.851063829787234,
821
+ "recall": 0.6808510638297872,
822
+ "f1": 0.7565011820330969,
823
+ "confusion_matrix": [
824
+ [
825
+ 791,
826
+ 28
827
+ ],
828
+ [
829
+ 75,
830
+ 160
831
+ ]
832
+ ],
833
+ "classification_report": {
834
+ "NOT_RELEVANT": {
835
+ "precision": 0.9133949191685913,
836
+ "recall": 0.9658119658119658,
837
+ "f1-score": 0.9388724035608309,
838
+ "support": 819.0
839
+ },
840
+ "RELEVANT": {
841
+ "precision": 0.851063829787234,
842
+ "recall": 0.6808510638297872,
843
+ "f1-score": 0.7565011820330969,
844
+ "support": 235.0
845
+ },
846
+ "accuracy": 0.9022770398481973,
847
+ "macro avg": {
848
+ "precision": 0.8822293744779126,
849
+ "recall": 0.8233315148208765,
850
+ "f1-score": 0.8476867927969639,
851
+ "support": 1054.0
852
+ },
853
+ "weighted avg": {
854
+ "precision": 0.8994975700180988,
855
+ "recall": 0.9022770398481973,
856
+ "f1-score": 0.8982108883245714,
857
+ "support": 1054.0
858
+ }
859
+ },
860
+ "roc_auc": 0.9321590938612216,
861
+ "average_precision": 0.8417078389566306
862
+ }
863
+ },
864
+ {
865
+ "model_type": "embedding-lightgbm_sentence_embeddings",
866
+ "model_name": "lightgbm",
867
+ "embedding_model_name": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
868
+ "artifact_dir": "/content/agri-maize_corn-classifier/baselines/embedding-lightgbm",
869
+ "artifact_file": "/content/agri-maize_corn-classifier/baselines/embedding-lightgbm/embedding-lightgbm.joblib",
870
+ "validation_best_threshold": {
871
+ "threshold": 0.14456141773750653,
872
+ "f1": 0.8134171907756813,
873
+ "precision": 0.7461538461538462,
874
+ "recall": 0.8940092165898618
875
+ },
876
+ "validation_default_0_5": {
877
+ "threshold": 0.5,
878
+ "accuracy": 0.9160377358490566,
879
+ "precision": 0.7909090909090909,
880
+ "recall": 0.8018433179723502,
881
+ "f1": 0.7963386727688787,
882
+ "confusion_matrix": [
883
+ [
884
+ 797,
885
+ 46
886
+ ],
887
+ [
888
+ 43,
889
+ 174
890
+ ]
891
+ ],
892
+ "classification_report": {
893
+ "NOT_RELEVANT": {
894
+ "precision": 0.9488095238095238,
895
+ "recall": 0.9454329774614472,
896
+ "f1-score": 0.9471182412358883,
897
+ "support": 843.0
898
+ },
899
+ "RELEVANT": {
900
+ "precision": 0.7909090909090909,
901
+ "recall": 0.8018433179723502,
902
+ "f1-score": 0.7963386727688787,
903
+ "support": 217.0
904
+ },
905
+ "accuracy": 0.9160377358490566,
906
+ "macro avg": {
907
+ "precision": 0.8698593073593073,
908
+ "recall": 0.8736381477168986,
909
+ "f1-score": 0.8717284570023835,
910
+ "support": 1060.0
911
+ },
912
+ "weighted avg": {
913
+ "precision": 0.9164846238666993,
914
+ "recall": 0.9160377358490566,
915
+ "f1-score": 0.916251103162925,
916
+ "support": 1060.0
917
+ }
918
+ },
919
+ "roc_auc": 0.9630571089645823,
920
+ "average_precision": 0.8780215936178333
921
+ },
922
+ "validation_optimal_threshold": {
923
+ "threshold": 0.14456141773750653,
924
+ "accuracy": 0.9160377358490566,
925
+ "precision": 0.7461538461538462,
926
+ "recall": 0.8940092165898618,
927
+ "f1": 0.8134171907756813,
928
+ "confusion_matrix": [
929
+ [
930
+ 777,
931
+ 66
932
+ ],
933
+ [
934
+ 23,
935
+ 194
936
+ ]
937
+ ],
938
+ "classification_report": {
939
+ "NOT_RELEVANT": {
940
+ "precision": 0.97125,
941
+ "recall": 0.9217081850533808,
942
+ "f1-score": 0.945830797321972,
943
+ "support": 843.0
944
+ },
945
+ "RELEVANT": {
946
+ "precision": 0.7461538461538462,
947
+ "recall": 0.8940092165898618,
948
+ "f1-score": 0.8134171907756813,
949
+ "support": 217.0
950
+ },
951
+ "accuracy": 0.9160377358490566,
952
+ "macro avg": {
953
+ "precision": 0.8587019230769231,
954
+ "recall": 0.9078587008216212,
955
+ "f1-score": 0.8796239940488266,
956
+ "support": 1060.0
957
+ },
958
+ "weighted avg": {
959
+ "precision": 0.9251689949201741,
960
+ "recall": 0.9160377358490566,
961
+ "f1-score": 0.9187234835290049,
962
+ "support": 1060.0
963
+ }
964
+ },
965
+ "roc_auc": 0.9630571089645823,
966
+ "average_precision": 0.8780215936178333
967
+ },
968
+ "test_default_0_5": {
969
+ "threshold": 0.5,
970
+ "accuracy": 0.9070208728652751,
971
+ "precision": 0.8702702702702703,
972
+ "recall": 0.6851063829787234,
973
+ "f1": 0.7666666666666667,
974
+ "confusion_matrix": [
975
+ [
976
+ 795,
977
+ 24
978
+ ],
979
+ [
980
+ 74,
981
+ 161
982
+ ]
983
+ ],
984
+ "classification_report": {
985
+ "NOT_RELEVANT": {
986
+ "precision": 0.9148446490218642,
987
+ "recall": 0.9706959706959707,
988
+ "f1-score": 0.9419431279620853,
989
+ "support": 819.0
990
+ },
991
+ "RELEVANT": {
992
+ "precision": 0.8702702702702703,
993
+ "recall": 0.6851063829787234,
994
+ "f1-score": 0.7666666666666667,
995
+ "support": 235.0
996
+ },
997
+ "accuracy": 0.9070208728652751,
998
+ "macro avg": {
999
+ "precision": 0.8925574596460673,
1000
+ "recall": 0.827901176837347,
1001
+ "f1-score": 0.8543048973143761,
1002
+ "support": 1054.0
1003
+ },
1004
+ "weighted avg": {
1005
+ "precision": 0.9049063387688998,
1006
+ "recall": 0.9070208728652751,
1007
+ "f1-score": 0.9028634615442264,
1008
+ "support": 1054.0
1009
+ }
1010
+ },
1011
+ "roc_auc": 0.9534149066063959,
1012
+ "average_precision": 0.872949726074902
1013
+ },
1014
+ "test_optimal_threshold": {
1015
+ "threshold": 0.14456141773750653,
1016
+ "accuracy": 0.9013282732447818,
1017
+ "precision": 0.7835497835497836,
1018
+ "recall": 0.7702127659574468,
1019
+ "f1": 0.776824034334764,
1020
+ "confusion_matrix": [
1021
+ [
1022
+ 769,
1023
+ 50
1024
+ ],
1025
+ [
1026
+ 54,
1027
+ 181
1028
+ ]
1029
+ ],
1030
+ "classification_report": {
1031
+ "NOT_RELEVANT": {
1032
+ "precision": 0.9343863912515188,
1033
+ "recall": 0.938949938949939,
1034
+ "f1-score": 0.9366626065773447,
1035
+ "support": 819.0
1036
+ },
1037
+ "RELEVANT": {
1038
+ "precision": 0.7835497835497836,
1039
+ "recall": 0.7702127659574468,
1040
+ "f1-score": 0.776824034334764,
1041
+ "support": 235.0
1042
+ },
1043
+ "accuracy": 0.9013282732447818,
1044
+ "macro avg": {
1045
+ "precision": 0.8589680874006511,
1046
+ "recall": 0.8545813524536929,
1047
+ "f1-score": 0.8567433204560544,
1048
+ "support": 1054.0
1049
+ },
1050
+ "weighted avg": {
1051
+ "precision": 0.9007558383009421,
1052
+ "recall": 0.9013282732447818,
1053
+ "f1-score": 0.9010249742462191,
1054
+ "support": 1054.0
1055
+ }
1056
+ },
1057
+ "roc_auc": 0.9534149066063959,
1058
+ "average_precision": 0.872949726074902
1059
+ }
1060
+ },
1061
+ {
1062
+ "model_type": "transformer",
1063
+ "model_name": "FacebookAI/xlm-roberta-base",
1064
+ "artifact_dir": "/content/agri-maize_corn-classifier/transformer",
1065
+ "validation_best_threshold": {
1066
+ "threshold": 0.32782965898513794,
1067
+ "f1": 0.9002320185614848,
1068
+ "precision": 0.9065420560747663,
1069
+ "recall": 0.8940092165898618
1070
+ },
1071
+ "validation_default_0_5": {
1072
+ "threshold": 0.5,
1073
+ "accuracy": 0.9575471698113207,
1074
+ "precision": 0.9134615384615384,
1075
+ "recall": 0.8755760368663594,
1076
+ "f1": 0.8941176470588236,
1077
+ "confusion_matrix": [
1078
+ [
1079
+ 825,
1080
+ 18
1081
+ ],
1082
+ [
1083
+ 27,
1084
+ 190
1085
+ ]
1086
+ ],
1087
+ "classification_report": {
1088
+ "NOT_RELEVANT": {
1089
+ "precision": 0.9683098591549296,
1090
+ "recall": 0.9786476868327402,
1091
+ "f1-score": 0.9734513274336283,
1092
+ "support": 843.0
1093
+ },
1094
+ "RELEVANT": {
1095
+ "precision": 0.9134615384615384,
1096
+ "recall": 0.8755760368663594,
1097
+ "f1-score": 0.8941176470588236,
1098
+ "support": 217.0
1099
+ },
1100
+ "accuracy": 0.9575471698113207,
1101
+ "macro avg": {
1102
+ "precision": 0.940885698808234,
1103
+ "recall": 0.9271118618495497,
1104
+ "f1-score": 0.933784487246226,
1105
+ "support": 1060.0
1106
+ },
1107
+ "weighted avg": {
1108
+ "precision": 0.9570814765224146,
1109
+ "recall": 0.9575471698113207,
1110
+ "f1-score": 0.9572103758852012,
1111
+ "support": 1060.0
1112
+ }
1113
+ },
1114
+ "roc_auc": 0.9729488167669778,
1115
+ "average_precision": 0.9434157534580784
1116
+ },
1117
+ "validation_optimal_threshold": {
1118
+ "threshold": 0.32782965898513794,
1119
+ "accuracy": 0.9594339622641509,
1120
+ "precision": 0.9065420560747663,
1121
+ "recall": 0.8940092165898618,
1122
+ "f1": 0.9002320185614849,
1123
+ "confusion_matrix": [
1124
+ [
1125
+ 823,
1126
+ 20
1127
+ ],
1128
+ [
1129
+ 23,
1130
+ 194
1131
+ ]
1132
+ ],
1133
+ "classification_report": {
1134
+ "NOT_RELEVANT": {
1135
+ "precision": 0.9728132387706856,
1136
+ "recall": 0.9762752075919335,
1137
+ "f1-score": 0.9745411486086442,
1138
+ "support": 843.0
1139
+ },
1140
+ "RELEVANT": {
1141
+ "precision": 0.9065420560747663,
1142
+ "recall": 0.8940092165898618,
1143
+ "f1-score": 0.9002320185614849,
1144
+ "support": 217.0
1145
+ },
1146
+ "accuracy": 0.9594339622641509,
1147
+ "macro avg": {
1148
+ "precision": 0.939677647422726,
1149
+ "recall": 0.9351422120908977,
1150
+ "f1-score": 0.9373865835850645,
1151
+ "support": 1060.0
1152
+ },
1153
+ "weighted avg": {
1154
+ "precision": 0.9592464023131247,
1155
+ "recall": 0.9594339622641509,
1156
+ "f1-score": 0.9593288078348389,
1157
+ "support": 1060.0
1158
+ }
1159
+ },
1160
+ "roc_auc": 0.9729488167669778,
1161
+ "average_precision": 0.9434157534580784
1162
+ },
1163
+ "test_default_0_5": {
1164
+ "threshold": 0.5,
1165
+ "accuracy": 0.9345351043643264,
1166
+ "precision": 0.8915094339622641,
1167
+ "recall": 0.8042553191489362,
1168
+ "f1": 0.8456375838926175,
1169
+ "confusion_matrix": [
1170
+ [
1171
+ 796,
1172
+ 23
1173
+ ],
1174
+ [
1175
+ 46,
1176
+ 189
1177
+ ]
1178
+ ],
1179
+ "classification_report": {
1180
+ "NOT_RELEVANT": {
1181
+ "precision": 0.9453681710213777,
1182
+ "recall": 0.9719169719169719,
1183
+ "f1-score": 0.9584587597832631,
1184
+ "support": 819.0
1185
+ },
1186
+ "RELEVANT": {
1187
+ "precision": 0.8915094339622641,
1188
+ "recall": 0.8042553191489362,
1189
+ "f1-score": 0.8456375838926175,
1190
+ "support": 235.0
1191
+ },
1192
+ "accuracy": 0.9345351043643264,
1193
+ "macro avg": {
1194
+ "precision": 0.9184388024918209,
1195
+ "recall": 0.8880861455329541,
1196
+ "f1-score": 0.9020481718379403,
1197
+ "support": 1054.0
1198
+ },
1199
+ "weighted avg": {
1200
+ "precision": 0.9333598188307785,
1201
+ "recall": 0.9345351043643264,
1202
+ "f1-score": 0.9333041332801306,
1203
+ "support": 1054.0
1204
+ }
1205
+ },
1206
+ "roc_auc": 0.9527134803730548,
1207
+ "average_precision": 0.8898057611290884
1208
+ },
1209
+ "test_optimal_threshold": {
1210
+ "threshold": 0.32782965898513794,
1211
+ "accuracy": 0.9345351043643264,
1212
+ "precision": 0.8807339449541285,
1213
+ "recall": 0.8170212765957446,
1214
+ "f1": 0.847682119205298,
1215
+ "confusion_matrix": [
1216
+ [
1217
+ 793,
1218
+ 26
1219
+ ],
1220
+ [
1221
+ 43,
1222
+ 192
1223
+ ]
1224
+ ],
1225
+ "classification_report": {
1226
+ "NOT_RELEVANT": {
1227
+ "precision": 0.9485645933014354,
1228
+ "recall": 0.9682539682539683,
1229
+ "f1-score": 0.9583081570996979,
1230
+ "support": 819.0
1231
+ },
1232
+ "RELEVANT": {
1233
+ "precision": 0.8807339449541285,
1234
+ "recall": 0.8170212765957446,
1235
+ "f1-score": 0.847682119205298,
1236
+ "support": 235.0
1237
+ },
1238
+ "accuracy": 0.9345351043643264,
1239
+ "macro avg": {
1240
+ "precision": 0.9146492691277819,
1241
+ "recall": 0.8926376224248564,
1242
+ "f1-score": 0.9029951381524979,
1243
+ "support": 1054.0
1244
+ },
1245
+ "weighted avg": {
1246
+ "precision": 0.9334410616490472,
1247
+ "recall": 0.9345351043643264,
1248
+ "f1-score": 0.9336429588974361,
1249
+ "support": 1054.0
1250
+ }
1251
+ },
1252
+ "roc_auc": 0.9527134803730548,
1253
+ "average_precision": 0.8898057611290884
1254
+ }
1255
+ }
1256
+ ]
1257
+ }
transformer/checkpoint-296/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "XLMRobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": 2,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "NOT_RELEVANT",
16
+ "1": "RELEVANT"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "is_decoder": false,
21
+ "label2id": {
22
+ "NOT_RELEVANT": 0,
23
+ "RELEVANT": 1
24
+ },
25
+ "layer_norm_eps": 1e-05,
26
+ "max_position_embeddings": 514,
27
+ "model_type": "xlm-roberta",
28
+ "num_attention_heads": 12,
29
+ "num_hidden_layers": 12,
30
+ "output_past": true,
31
+ "pad_token_id": 1,
32
+ "position_embedding_type": "absolute",
33
+ "problem_type": "single_label_classification",
34
+ "tie_word_embeddings": true,
35
+ "transformers_version": "5.10.2",
36
+ "type_vocab_size": 1,
37
+ "use_cache": false,
38
+ "vocab_size": 250002
39
+ }
transformer/checkpoint-296/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85e55690177f9df9681c79755fefed0d01c49b690d5fe3f1e4e5e2e1fd2051e9
3
+ size 1112205008
transformer/checkpoint-296/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfab16a6ff59a8be730741b355132a64df206f043501efe631b370c845f2538c
3
+ size 2224532875
transformer/checkpoint-296/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15581d46ce8805f1dfa758ecf9798fec9b05fc7a0f3e100c4f3a339fe05fee89
3
+ size 14645
transformer/checkpoint-296/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bdd9077530e85b1b9e5cf984a9ffda5e212c37f58d6ee425f3578887c7b014b
3
+ size 1383
transformer/checkpoint-296/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dea5a43be8a53415ef4dd342591c550950514872b62cacca5fe3d312f57423e2
3
+ size 1465
transformer/checkpoint-296/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc02d42fb2a10276563109e2287cc0dbe6b595d5b3b3401c7cfeffc0b7e20270
3
+ size 17098351
transformer/checkpoint-296/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<s>",
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "mask_token": "<mask>",
10
+ "model_max_length": 512,
11
+ "pad_token": "<pad>",
12
+ "sep_token": "</s>",
13
+ "tokenizer_class": "XLMRobertaTokenizer",
14
+ "unk_token": "<unk>"
15
+ }
transformer/checkpoint-296/trainer_state.json ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 296,
3
+ "best_metric": 0.8941176470588236,
4
+ "best_model_checkpoint": "/content/agri-maize_corn-classifier/transformer/checkpoint-296",
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 296,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.08445945945945946,
14
+ "grad_norm": 4.2122979164123535,
15
+ "learning_rate": 3.2432432432432437e-06,
16
+ "loss": 0.6247027587890625,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.16891891891891891,
21
+ "grad_norm": 7.709840297698975,
22
+ "learning_rate": 6.621621621621622e-06,
23
+ "loss": 0.5710235595703125,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.2533783783783784,
28
+ "grad_norm": 6.6395344734191895,
29
+ "learning_rate": 1e-05,
30
+ "loss": 0.5104244995117188,
31
+ "step": 75
32
+ },
33
+ {
34
+ "epoch": 0.33783783783783783,
35
+ "grad_norm": 5.032057762145996,
36
+ "learning_rate": 1.3378378378378381e-05,
37
+ "loss": 0.39651737213134763,
38
+ "step": 100
39
+ },
40
+ {
41
+ "epoch": 0.4222972972972973,
42
+ "grad_norm": 26.106103897094727,
43
+ "learning_rate": 1.6756756756756757e-05,
44
+ "loss": 0.30904300689697267,
45
+ "step": 125
46
+ },
47
+ {
48
+ "epoch": 0.5067567567567568,
49
+ "grad_norm": 19.755094528198242,
50
+ "learning_rate": 1.9984984984984987e-05,
51
+ "loss": 0.21882522583007813,
52
+ "step": 150
53
+ },
54
+ {
55
+ "epoch": 0.5912162162162162,
56
+ "grad_norm": 10.216408729553223,
57
+ "learning_rate": 1.9609609609609613e-05,
58
+ "loss": 0.2193868064880371,
59
+ "step": 175
60
+ },
61
+ {
62
+ "epoch": 0.6756756756756757,
63
+ "grad_norm": 19.423349380493164,
64
+ "learning_rate": 1.9234234234234235e-05,
65
+ "loss": 0.2879533958435059,
66
+ "step": 200
67
+ },
68
+ {
69
+ "epoch": 0.7601351351351351,
70
+ "grad_norm": 3.9074134826660156,
71
+ "learning_rate": 1.885885885885886e-05,
72
+ "loss": 0.21735734939575196,
73
+ "step": 225
74
+ },
75
+ {
76
+ "epoch": 0.8445945945945946,
77
+ "grad_norm": 0.8604102730751038,
78
+ "learning_rate": 1.8483483483483483e-05,
79
+ "loss": 0.28983583450317385,
80
+ "step": 250
81
+ },
82
+ {
83
+ "epoch": 0.9290540540540541,
84
+ "grad_norm": 49.190547943115234,
85
+ "learning_rate": 1.8108108108108108e-05,
86
+ "loss": 0.2321755599975586,
87
+ "step": 275
88
+ },
89
+ {
90
+ "epoch": 1.0,
91
+ "eval_accuracy": 0.9575471698113207,
92
+ "eval_f1": 0.8941176470588236,
93
+ "eval_loss": 0.16295333206653595,
94
+ "eval_precision": 0.9134615384615384,
95
+ "eval_recall": 0.8755760368663594,
96
+ "eval_roc_auc": 0.9729488167669778,
97
+ "eval_runtime": 3.9863,
98
+ "eval_samples_per_second": 265.91,
99
+ "eval_steps_per_second": 8.529,
100
+ "step": 296
101
+ }
102
+ ],
103
+ "logging_steps": 25,
104
+ "max_steps": 1480,
105
+ "num_input_tokens_seen": 0,
106
+ "num_train_epochs": 5,
107
+ "save_steps": 500,
108
+ "stateful_callbacks": {
109
+ "EarlyStoppingCallback": {
110
+ "args": {
111
+ "early_stopping_patience": 2,
112
+ "early_stopping_threshold": 0.0
113
+ },
114
+ "attributes": {
115
+ "early_stopping_patience_counter": 0
116
+ }
117
+ },
118
+ "TrainerControl": {
119
+ "args": {
120
+ "should_epoch_stop": false,
121
+ "should_evaluate": false,
122
+ "should_log": false,
123
+ "should_save": true,
124
+ "should_training_stop": false
125
+ },
126
+ "attributes": {}
127
+ }
128
+ },
129
+ "total_flos": 621468312760320.0,
130
+ "train_batch_size": 16,
131
+ "trial_name": null,
132
+ "trial_params": null
133
+ }
transformer/checkpoint-296/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:699ccdec1f51a0096482e3403ced3b8ee41aac4e70941e5bd8752f8c52f7c09a
3
+ size 5201
transformer/checkpoint-592/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "XLMRobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": 2,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "NOT_RELEVANT",
16
+ "1": "RELEVANT"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "is_decoder": false,
21
+ "label2id": {
22
+ "NOT_RELEVANT": 0,
23
+ "RELEVANT": 1
24
+ },
25
+ "layer_norm_eps": 1e-05,
26
+ "max_position_embeddings": 514,
27
+ "model_type": "xlm-roberta",
28
+ "num_attention_heads": 12,
29
+ "num_hidden_layers": 12,
30
+ "output_past": true,
31
+ "pad_token_id": 1,
32
+ "position_embedding_type": "absolute",
33
+ "problem_type": "single_label_classification",
34
+ "tie_word_embeddings": true,
35
+ "transformers_version": "5.10.2",
36
+ "type_vocab_size": 1,
37
+ "use_cache": false,
38
+ "vocab_size": 250002
39
+ }
transformer/checkpoint-592/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c8c813341f964cfd0325ef4c81479a207e5a21073dab530cdc25e79b888fb63
3
+ size 1112205008
transformer/checkpoint-592/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b1916c2448623e28c5421dd95599e8383f9fcce2d4426633b740d2fddbd7df8
3
+ size 2224532875
transformer/checkpoint-592/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17d83d8dd15f1fd53bee866bfc712bc4a0cdfe918ef088a873e4be8cec157e21
3
+ size 14645
transformer/checkpoint-592/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d96563bcdba4b0d6b938c6748c5982ab5b9a030136631e3cfe83646b6d548c9
3
+ size 1383
transformer/checkpoint-592/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04c7c10124c7753e91192eb280101f3ff7bcb30733117a564929fe9ac6609ca4
3
+ size 1465
transformer/checkpoint-592/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc02d42fb2a10276563109e2287cc0dbe6b595d5b3b3401c7cfeffc0b7e20270
3
+ size 17098351
transformer/checkpoint-592/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<s>",
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "mask_token": "<mask>",
10
+ "model_max_length": 512,
11
+ "pad_token": "<pad>",
12
+ "sep_token": "</s>",
13
+ "tokenizer_class": "XLMRobertaTokenizer",
14
+ "unk_token": "<unk>"
15
+ }
transformer/checkpoint-592/trainer_state.json ADDED
@@ -0,0 +1,230 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 296,
3
+ "best_metric": 0.8941176470588236,
4
+ "best_model_checkpoint": "/content/agri-maize_corn-classifier/transformer/checkpoint-296",
5
+ "epoch": 2.0,
6
+ "eval_steps": 500,
7
+ "global_step": 592,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.08445945945945946,
14
+ "grad_norm": 4.2122979164123535,
15
+ "learning_rate": 3.2432432432432437e-06,
16
+ "loss": 0.6247027587890625,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.16891891891891891,
21
+ "grad_norm": 7.709840297698975,
22
+ "learning_rate": 6.621621621621622e-06,
23
+ "loss": 0.5710235595703125,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.2533783783783784,
28
+ "grad_norm": 6.6395344734191895,
29
+ "learning_rate": 1e-05,
30
+ "loss": 0.5104244995117188,
31
+ "step": 75
32
+ },
33
+ {
34
+ "epoch": 0.33783783783783783,
35
+ "grad_norm": 5.032057762145996,
36
+ "learning_rate": 1.3378378378378381e-05,
37
+ "loss": 0.39651737213134763,
38
+ "step": 100
39
+ },
40
+ {
41
+ "epoch": 0.4222972972972973,
42
+ "grad_norm": 26.106103897094727,
43
+ "learning_rate": 1.6756756756756757e-05,
44
+ "loss": 0.30904300689697267,
45
+ "step": 125
46
+ },
47
+ {
48
+ "epoch": 0.5067567567567568,
49
+ "grad_norm": 19.755094528198242,
50
+ "learning_rate": 1.9984984984984987e-05,
51
+ "loss": 0.21882522583007813,
52
+ "step": 150
53
+ },
54
+ {
55
+ "epoch": 0.5912162162162162,
56
+ "grad_norm": 10.216408729553223,
57
+ "learning_rate": 1.9609609609609613e-05,
58
+ "loss": 0.2193868064880371,
59
+ "step": 175
60
+ },
61
+ {
62
+ "epoch": 0.6756756756756757,
63
+ "grad_norm": 19.423349380493164,
64
+ "learning_rate": 1.9234234234234235e-05,
65
+ "loss": 0.2879533958435059,
66
+ "step": 200
67
+ },
68
+ {
69
+ "epoch": 0.7601351351351351,
70
+ "grad_norm": 3.9074134826660156,
71
+ "learning_rate": 1.885885885885886e-05,
72
+ "loss": 0.21735734939575196,
73
+ "step": 225
74
+ },
75
+ {
76
+ "epoch": 0.8445945945945946,
77
+ "grad_norm": 0.8604102730751038,
78
+ "learning_rate": 1.8483483483483483e-05,
79
+ "loss": 0.28983583450317385,
80
+ "step": 250
81
+ },
82
+ {
83
+ "epoch": 0.9290540540540541,
84
+ "grad_norm": 49.190547943115234,
85
+ "learning_rate": 1.8108108108108108e-05,
86
+ "loss": 0.2321755599975586,
87
+ "step": 275
88
+ },
89
+ {
90
+ "epoch": 1.0,
91
+ "eval_accuracy": 0.9575471698113207,
92
+ "eval_f1": 0.8941176470588236,
93
+ "eval_loss": 0.16295333206653595,
94
+ "eval_precision": 0.9134615384615384,
95
+ "eval_recall": 0.8755760368663594,
96
+ "eval_roc_auc": 0.9729488167669778,
97
+ "eval_runtime": 3.9863,
98
+ "eval_samples_per_second": 265.91,
99
+ "eval_steps_per_second": 8.529,
100
+ "step": 296
101
+ },
102
+ {
103
+ "epoch": 1.0135135135135136,
104
+ "grad_norm": 0.8517799377441406,
105
+ "learning_rate": 1.7732732732732734e-05,
106
+ "loss": 0.18440502166748046,
107
+ "step": 300
108
+ },
109
+ {
110
+ "epoch": 1.097972972972973,
111
+ "grad_norm": 2.9610507488250732,
112
+ "learning_rate": 1.735735735735736e-05,
113
+ "loss": 0.16707677841186525,
114
+ "step": 325
115
+ },
116
+ {
117
+ "epoch": 1.1824324324324325,
118
+ "grad_norm": 4.359492301940918,
119
+ "learning_rate": 1.6981981981981985e-05,
120
+ "loss": 0.22817501068115234,
121
+ "step": 350
122
+ },
123
+ {
124
+ "epoch": 1.2668918918918919,
125
+ "grad_norm": 0.49442264437675476,
126
+ "learning_rate": 1.6606606606606607e-05,
127
+ "loss": 0.16232917785644532,
128
+ "step": 375
129
+ },
130
+ {
131
+ "epoch": 1.3513513513513513,
132
+ "grad_norm": 0.9959002733230591,
133
+ "learning_rate": 1.6231231231231232e-05,
134
+ "loss": 0.2139061164855957,
135
+ "step": 400
136
+ },
137
+ {
138
+ "epoch": 1.4358108108108107,
139
+ "grad_norm": 1.0536309480667114,
140
+ "learning_rate": 1.5855855855855858e-05,
141
+ "loss": 0.203138484954834,
142
+ "step": 425
143
+ },
144
+ {
145
+ "epoch": 1.5202702702702702,
146
+ "grad_norm": 2.8971381187438965,
147
+ "learning_rate": 1.5480480480480483e-05,
148
+ "loss": 0.18809621810913085,
149
+ "step": 450
150
+ },
151
+ {
152
+ "epoch": 1.6047297297297298,
153
+ "grad_norm": 0.34616008400917053,
154
+ "learning_rate": 1.5105105105105107e-05,
155
+ "loss": 0.15637354850769042,
156
+ "step": 475
157
+ },
158
+ {
159
+ "epoch": 1.689189189189189,
160
+ "grad_norm": 10.969264030456543,
161
+ "learning_rate": 1.4729729729729731e-05,
162
+ "loss": 0.22628171920776366,
163
+ "step": 500
164
+ },
165
+ {
166
+ "epoch": 1.7736486486486487,
167
+ "grad_norm": 17.27104377746582,
168
+ "learning_rate": 1.4354354354354357e-05,
169
+ "loss": 0.2247269058227539,
170
+ "step": 525
171
+ },
172
+ {
173
+ "epoch": 1.8581081081081081,
174
+ "grad_norm": 1.122333288192749,
175
+ "learning_rate": 1.397897897897898e-05,
176
+ "loss": 0.1473069667816162,
177
+ "step": 550
178
+ },
179
+ {
180
+ "epoch": 1.9425675675675675,
181
+ "grad_norm": 0.30135178565979004,
182
+ "learning_rate": 1.3603603603603606e-05,
183
+ "loss": 0.18822559356689453,
184
+ "step": 575
185
+ },
186
+ {
187
+ "epoch": 2.0,
188
+ "eval_accuracy": 0.9471698113207547,
189
+ "eval_f1": 0.8771929824561403,
190
+ "eval_loss": 0.19445720314979553,
191
+ "eval_precision": 0.8368200836820083,
192
+ "eval_recall": 0.9216589861751152,
193
+ "eval_roc_auc": 0.9843055578332813,
194
+ "eval_runtime": 3.9078,
195
+ "eval_samples_per_second": 271.255,
196
+ "eval_steps_per_second": 8.701,
197
+ "step": 592
198
+ }
199
+ ],
200
+ "logging_steps": 25,
201
+ "max_steps": 1480,
202
+ "num_input_tokens_seen": 0,
203
+ "num_train_epochs": 5,
204
+ "save_steps": 500,
205
+ "stateful_callbacks": {
206
+ "EarlyStoppingCallback": {
207
+ "args": {
208
+ "early_stopping_patience": 2,
209
+ "early_stopping_threshold": 0.0
210
+ },
211
+ "attributes": {
212
+ "early_stopping_patience_counter": 1
213
+ }
214
+ },
215
+ "TrainerControl": {
216
+ "args": {
217
+ "should_epoch_stop": false,
218
+ "should_evaluate": false,
219
+ "should_log": false,
220
+ "should_save": true,
221
+ "should_training_stop": false
222
+ },
223
+ "attributes": {}
224
+ }
225
+ },
226
+ "total_flos": 1242936625520640.0,
227
+ "train_batch_size": 16,
228
+ "trial_name": null,
229
+ "trial_params": null
230
+ }
transformer/checkpoint-592/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:699ccdec1f51a0096482e3403ced3b8ee41aac4e70941e5bd8752f8c52f7c09a
3
+ size 5201
transformer/checkpoint-888/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "XLMRobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": 2,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "NOT_RELEVANT",
16
+ "1": "RELEVANT"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "is_decoder": false,
21
+ "label2id": {
22
+ "NOT_RELEVANT": 0,
23
+ "RELEVANT": 1
24
+ },
25
+ "layer_norm_eps": 1e-05,
26
+ "max_position_embeddings": 514,
27
+ "model_type": "xlm-roberta",
28
+ "num_attention_heads": 12,
29
+ "num_hidden_layers": 12,
30
+ "output_past": true,
31
+ "pad_token_id": 1,
32
+ "position_embedding_type": "absolute",
33
+ "problem_type": "single_label_classification",
34
+ "tie_word_embeddings": true,
35
+ "transformers_version": "5.10.2",
36
+ "type_vocab_size": 1,
37
+ "use_cache": false,
38
+ "vocab_size": 250002
39
+ }
transformer/checkpoint-888/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49fc7b3e1231b0df1ee6197d4b79cc977ee6efe769601a28d08f6155e0394429
3
+ size 1112205008
transformer/checkpoint-888/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82d7fa175c5005a01599e4fdec14598bd8f76dcd60ecf72e216aaf04e3b735b7
3
+ size 2224532875
transformer/checkpoint-888/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:637187f6bca9c312460a048bc23889499d68e9152f595a188749add4b6bb4354
3
+ size 14645
transformer/checkpoint-888/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:626d811fdb20e63a2a61563727ae0fb44866597b95cbddab1196a4cb5e10b6cf
3
+ size 1383
transformer/checkpoint-888/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:012e4ef245198acca779ad61b1e34757abaaba4f83c3045c244728e9fce24daf
3
+ size 1465
transformer/checkpoint-888/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc02d42fb2a10276563109e2287cc0dbe6b595d5b3b3401c7cfeffc0b7e20270
3
+ size 17098351
transformer/checkpoint-888/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<s>",
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "mask_token": "<mask>",
10
+ "model_max_length": 512,
11
+ "pad_token": "<pad>",
12
+ "sep_token": "</s>",
13
+ "tokenizer_class": "XLMRobertaTokenizer",
14
+ "unk_token": "<unk>"
15
+ }
transformer/checkpoint-888/trainer_state.json ADDED
@@ -0,0 +1,327 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 296,
3
+ "best_metric": 0.8941176470588236,
4
+ "best_model_checkpoint": "/content/agri-maize_corn-classifier/transformer/checkpoint-296",
5
+ "epoch": 3.0,
6
+ "eval_steps": 500,
7
+ "global_step": 888,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.08445945945945946,
14
+ "grad_norm": 4.2122979164123535,
15
+ "learning_rate": 3.2432432432432437e-06,
16
+ "loss": 0.6247027587890625,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.16891891891891891,
21
+ "grad_norm": 7.709840297698975,
22
+ "learning_rate": 6.621621621621622e-06,
23
+ "loss": 0.5710235595703125,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.2533783783783784,
28
+ "grad_norm": 6.6395344734191895,
29
+ "learning_rate": 1e-05,
30
+ "loss": 0.5104244995117188,
31
+ "step": 75
32
+ },
33
+ {
34
+ "epoch": 0.33783783783783783,
35
+ "grad_norm": 5.032057762145996,
36
+ "learning_rate": 1.3378378378378381e-05,
37
+ "loss": 0.39651737213134763,
38
+ "step": 100
39
+ },
40
+ {
41
+ "epoch": 0.4222972972972973,
42
+ "grad_norm": 26.106103897094727,
43
+ "learning_rate": 1.6756756756756757e-05,
44
+ "loss": 0.30904300689697267,
45
+ "step": 125
46
+ },
47
+ {
48
+ "epoch": 0.5067567567567568,
49
+ "grad_norm": 19.755094528198242,
50
+ "learning_rate": 1.9984984984984987e-05,
51
+ "loss": 0.21882522583007813,
52
+ "step": 150
53
+ },
54
+ {
55
+ "epoch": 0.5912162162162162,
56
+ "grad_norm": 10.216408729553223,
57
+ "learning_rate": 1.9609609609609613e-05,
58
+ "loss": 0.2193868064880371,
59
+ "step": 175
60
+ },
61
+ {
62
+ "epoch": 0.6756756756756757,
63
+ "grad_norm": 19.423349380493164,
64
+ "learning_rate": 1.9234234234234235e-05,
65
+ "loss": 0.2879533958435059,
66
+ "step": 200
67
+ },
68
+ {
69
+ "epoch": 0.7601351351351351,
70
+ "grad_norm": 3.9074134826660156,
71
+ "learning_rate": 1.885885885885886e-05,
72
+ "loss": 0.21735734939575196,
73
+ "step": 225
74
+ },
75
+ {
76
+ "epoch": 0.8445945945945946,
77
+ "grad_norm": 0.8604102730751038,
78
+ "learning_rate": 1.8483483483483483e-05,
79
+ "loss": 0.28983583450317385,
80
+ "step": 250
81
+ },
82
+ {
83
+ "epoch": 0.9290540540540541,
84
+ "grad_norm": 49.190547943115234,
85
+ "learning_rate": 1.8108108108108108e-05,
86
+ "loss": 0.2321755599975586,
87
+ "step": 275
88
+ },
89
+ {
90
+ "epoch": 1.0,
91
+ "eval_accuracy": 0.9575471698113207,
92
+ "eval_f1": 0.8941176470588236,
93
+ "eval_loss": 0.16295333206653595,
94
+ "eval_precision": 0.9134615384615384,
95
+ "eval_recall": 0.8755760368663594,
96
+ "eval_roc_auc": 0.9729488167669778,
97
+ "eval_runtime": 3.9863,
98
+ "eval_samples_per_second": 265.91,
99
+ "eval_steps_per_second": 8.529,
100
+ "step": 296
101
+ },
102
+ {
103
+ "epoch": 1.0135135135135136,
104
+ "grad_norm": 0.8517799377441406,
105
+ "learning_rate": 1.7732732732732734e-05,
106
+ "loss": 0.18440502166748046,
107
+ "step": 300
108
+ },
109
+ {
110
+ "epoch": 1.097972972972973,
111
+ "grad_norm": 2.9610507488250732,
112
+ "learning_rate": 1.735735735735736e-05,
113
+ "loss": 0.16707677841186525,
114
+ "step": 325
115
+ },
116
+ {
117
+ "epoch": 1.1824324324324325,
118
+ "grad_norm": 4.359492301940918,
119
+ "learning_rate": 1.6981981981981985e-05,
120
+ "loss": 0.22817501068115234,
121
+ "step": 350
122
+ },
123
+ {
124
+ "epoch": 1.2668918918918919,
125
+ "grad_norm": 0.49442264437675476,
126
+ "learning_rate": 1.6606606606606607e-05,
127
+ "loss": 0.16232917785644532,
128
+ "step": 375
129
+ },
130
+ {
131
+ "epoch": 1.3513513513513513,
132
+ "grad_norm": 0.9959002733230591,
133
+ "learning_rate": 1.6231231231231232e-05,
134
+ "loss": 0.2139061164855957,
135
+ "step": 400
136
+ },
137
+ {
138
+ "epoch": 1.4358108108108107,
139
+ "grad_norm": 1.0536309480667114,
140
+ "learning_rate": 1.5855855855855858e-05,
141
+ "loss": 0.203138484954834,
142
+ "step": 425
143
+ },
144
+ {
145
+ "epoch": 1.5202702702702702,
146
+ "grad_norm": 2.8971381187438965,
147
+ "learning_rate": 1.5480480480480483e-05,
148
+ "loss": 0.18809621810913085,
149
+ "step": 450
150
+ },
151
+ {
152
+ "epoch": 1.6047297297297298,
153
+ "grad_norm": 0.34616008400917053,
154
+ "learning_rate": 1.5105105105105107e-05,
155
+ "loss": 0.15637354850769042,
156
+ "step": 475
157
+ },
158
+ {
159
+ "epoch": 1.689189189189189,
160
+ "grad_norm": 10.969264030456543,
161
+ "learning_rate": 1.4729729729729731e-05,
162
+ "loss": 0.22628171920776366,
163
+ "step": 500
164
+ },
165
+ {
166
+ "epoch": 1.7736486486486487,
167
+ "grad_norm": 17.27104377746582,
168
+ "learning_rate": 1.4354354354354357e-05,
169
+ "loss": 0.2247269058227539,
170
+ "step": 525
171
+ },
172
+ {
173
+ "epoch": 1.8581081081081081,
174
+ "grad_norm": 1.122333288192749,
175
+ "learning_rate": 1.397897897897898e-05,
176
+ "loss": 0.1473069667816162,
177
+ "step": 550
178
+ },
179
+ {
180
+ "epoch": 1.9425675675675675,
181
+ "grad_norm": 0.30135178565979004,
182
+ "learning_rate": 1.3603603603603606e-05,
183
+ "loss": 0.18822559356689453,
184
+ "step": 575
185
+ },
186
+ {
187
+ "epoch": 2.0,
188
+ "eval_accuracy": 0.9471698113207547,
189
+ "eval_f1": 0.8771929824561403,
190
+ "eval_loss": 0.19445720314979553,
191
+ "eval_precision": 0.8368200836820083,
192
+ "eval_recall": 0.9216589861751152,
193
+ "eval_roc_auc": 0.9843055578332813,
194
+ "eval_runtime": 3.9078,
195
+ "eval_samples_per_second": 271.255,
196
+ "eval_steps_per_second": 8.701,
197
+ "step": 592
198
+ },
199
+ {
200
+ "epoch": 2.027027027027027,
201
+ "grad_norm": 26.52724838256836,
202
+ "learning_rate": 1.322822822822823e-05,
203
+ "loss": 0.13748784065246583,
204
+ "step": 600
205
+ },
206
+ {
207
+ "epoch": 2.1114864864864864,
208
+ "grad_norm": 10.8816499710083,
209
+ "learning_rate": 1.2852852852852854e-05,
210
+ "loss": 0.1081045150756836,
211
+ "step": 625
212
+ },
213
+ {
214
+ "epoch": 2.195945945945946,
215
+ "grad_norm": 2.227597713470459,
216
+ "learning_rate": 1.2477477477477477e-05,
217
+ "loss": 0.14378289222717286,
218
+ "step": 650
219
+ },
220
+ {
221
+ "epoch": 2.2804054054054053,
222
+ "grad_norm": 12.489811897277832,
223
+ "learning_rate": 1.2102102102102103e-05,
224
+ "loss": 0.1484208297729492,
225
+ "step": 675
226
+ },
227
+ {
228
+ "epoch": 2.364864864864865,
229
+ "grad_norm": 28.363985061645508,
230
+ "learning_rate": 1.1726726726726727e-05,
231
+ "loss": 0.09925691604614258,
232
+ "step": 700
233
+ },
234
+ {
235
+ "epoch": 2.4493243243243246,
236
+ "grad_norm": 10.956056594848633,
237
+ "learning_rate": 1.1351351351351352e-05,
238
+ "loss": 0.1072914218902588,
239
+ "step": 725
240
+ },
241
+ {
242
+ "epoch": 2.5337837837837838,
243
+ "grad_norm": 35.35126876831055,
244
+ "learning_rate": 1.0975975975975976e-05,
245
+ "loss": 0.16145267486572265,
246
+ "step": 750
247
+ },
248
+ {
249
+ "epoch": 2.618243243243243,
250
+ "grad_norm": 4.4620256423950195,
251
+ "learning_rate": 1.0600600600600602e-05,
252
+ "loss": 0.12065947532653809,
253
+ "step": 775
254
+ },
255
+ {
256
+ "epoch": 2.7027027027027026,
257
+ "grad_norm": 3.3222928047180176,
258
+ "learning_rate": 1.0225225225225226e-05,
259
+ "loss": 0.1423185920715332,
260
+ "step": 800
261
+ },
262
+ {
263
+ "epoch": 2.7871621621621623,
264
+ "grad_norm": 0.3517614006996155,
265
+ "learning_rate": 9.849849849849851e-06,
266
+ "loss": 0.14739763259887695,
267
+ "step": 825
268
+ },
269
+ {
270
+ "epoch": 2.8716216216216215,
271
+ "grad_norm": 8.386738777160645,
272
+ "learning_rate": 9.474474474474475e-06,
273
+ "loss": 0.14476879119873046,
274
+ "step": 850
275
+ },
276
+ {
277
+ "epoch": 2.956081081081081,
278
+ "grad_norm": 1.721596121788025,
279
+ "learning_rate": 9.0990990990991e-06,
280
+ "loss": 0.10418204307556152,
281
+ "step": 875
282
+ },
283
+ {
284
+ "epoch": 3.0,
285
+ "eval_accuracy": 0.9471698113207547,
286
+ "eval_f1": 0.8761061946902655,
287
+ "eval_loss": 0.20338048040866852,
288
+ "eval_precision": 0.8425531914893617,
289
+ "eval_recall": 0.9124423963133641,
290
+ "eval_roc_auc": 0.9789784126255255,
291
+ "eval_runtime": 3.8956,
292
+ "eval_samples_per_second": 272.099,
293
+ "eval_steps_per_second": 8.728,
294
+ "step": 888
295
+ }
296
+ ],
297
+ "logging_steps": 25,
298
+ "max_steps": 1480,
299
+ "num_input_tokens_seen": 0,
300
+ "num_train_epochs": 5,
301
+ "save_steps": 500,
302
+ "stateful_callbacks": {
303
+ "EarlyStoppingCallback": {
304
+ "args": {
305
+ "early_stopping_patience": 2,
306
+ "early_stopping_threshold": 0.0
307
+ },
308
+ "attributes": {
309
+ "early_stopping_patience_counter": 2
310
+ }
311
+ },
312
+ "TrainerControl": {
313
+ "args": {
314
+ "should_epoch_stop": false,
315
+ "should_evaluate": false,
316
+ "should_log": false,
317
+ "should_save": true,
318
+ "should_training_stop": true
319
+ },
320
+ "attributes": {}
321
+ }
322
+ },
323
+ "total_flos": 1864404938280960.0,
324
+ "train_batch_size": 16,
325
+ "trial_name": null,
326
+ "trial_params": null
327
+ }
transformer/checkpoint-888/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:699ccdec1f51a0096482e3403ced3b8ee41aac4e70941e5bd8752f8c52f7c09a
3
+ size 5201
transformer/config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "XLMRobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": 2,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "NOT_RELEVANT",
16
+ "1": "RELEVANT"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "is_decoder": false,
21
+ "label2id": {
22
+ "NOT_RELEVANT": 0,
23
+ "RELEVANT": 1
24
+ },
25
+ "layer_norm_eps": 1e-05,
26
+ "max_position_embeddings": 514,
27
+ "model_type": "xlm-roberta",
28
+ "num_attention_heads": 12,
29
+ "num_hidden_layers": 12,
30
+ "output_past": true,
31
+ "pad_token_id": 1,
32
+ "position_embedding_type": "absolute",
33
+ "problem_type": "single_label_classification",
34
+ "threshold": 0.32782965898513794,
35
+ "tie_word_embeddings": true,
36
+ "transformers_version": "5.10.2",
37
+ "type_vocab_size": 1,
38
+ "use_cache": false,
39
+ "validation_threshold_report": {
40
+ "f1": 0.9002320185614848,
41
+ "precision": 0.9065420560747663,
42
+ "recall": 0.8940092165898618,
43
+ "threshold": 0.32782965898513794
44
+ },
45
+ "vocab_size": 250002
46
+ }