karths commited on
Commit
37c4901
·
verified ·
1 Parent(s): 554f8cb

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ 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
+ reduced_main_data.csv filter=lfs diff=lfs merge=lfs -text
37
+ test_data_for_future_evaluation.csv filter=lfs diff=lfs merge=lfs -text
checkpoint-39015/config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilroberta-base",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 6,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
+ "torch_dtype": "float32",
24
+ "transformers_version": "4.35.0",
25
+ "type_vocab_size": 1,
26
+ "use_cache": true,
27
+ "vocab_size": 50265
28
+ }
checkpoint-39015/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1b58662fe2ed9acfb01ffced17a3680cc0708c8a5107b653a9ce8c497179458
3
+ size 328492280
checkpoint-39015/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba64fcd226ef55d2905eaf4977b25772211e1e9a068848369730f7d0a1610f7f
3
+ size 657047610
checkpoint-39015/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9adb5bd3a062ebd1d21c9334d6d8b85df023540f771a086f57cc8450b18fd3f
3
+ size 14244
checkpoint-39015/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd8176fd6618dd77e680121cd0ef96d88b28e83690ad66ba643e81bdcada5cb5
3
+ size 1064
checkpoint-39015/trainer_state.json ADDED
@@ -0,0 +1,487 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.999679630934837,
5
+ "eval_steps": 500,
6
+ "global_step": 39015,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.06,
13
+ "learning_rate": 4.9800000000000004e-05,
14
+ "loss": 0.049,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 0.13,
19
+ "learning_rate": 4.997937713868709e-05,
20
+ "loss": 0.0574,
21
+ "step": 1000
22
+ },
23
+ {
24
+ "epoch": 0.19,
25
+ "learning_rate": 4.991721127523219e-05,
26
+ "loss": 0.0621,
27
+ "step": 1500
28
+ },
29
+ {
30
+ "epoch": 0.26,
31
+ "learning_rate": 4.981360546534587e-05,
32
+ "loss": 0.0639,
33
+ "step": 2000
34
+ },
35
+ {
36
+ "epoch": 0.32,
37
+ "learning_rate": 4.966873201640091e-05,
38
+ "loss": 0.0679,
39
+ "step": 2500
40
+ },
41
+ {
42
+ "epoch": 0.38,
43
+ "learning_rate": 4.948283186819981e-05,
44
+ "loss": 0.0746,
45
+ "step": 3000
46
+ },
47
+ {
48
+ "epoch": 0.45,
49
+ "learning_rate": 4.9256214192266534e-05,
50
+ "loss": 0.0778,
51
+ "step": 3500
52
+ },
53
+ {
54
+ "epoch": 0.51,
55
+ "learning_rate": 4.898982977024972e-05,
56
+ "loss": 0.0752,
57
+ "step": 4000
58
+ },
59
+ {
60
+ "epoch": 0.58,
61
+ "learning_rate": 4.868305410275411e-05,
62
+ "loss": 0.0763,
63
+ "step": 4500
64
+ },
65
+ {
66
+ "epoch": 0.64,
67
+ "learning_rate": 4.833689102217645e-05,
68
+ "loss": 0.0769,
69
+ "step": 5000
70
+ },
71
+ {
72
+ "epoch": 0.7,
73
+ "learning_rate": 4.795191623417485e-05,
74
+ "loss": 0.0755,
75
+ "step": 5500
76
+ },
77
+ {
78
+ "epoch": 0.77,
79
+ "learning_rate": 4.752965392266396e-05,
80
+ "loss": 0.0751,
81
+ "step": 6000
82
+ },
83
+ {
84
+ "epoch": 0.83,
85
+ "learning_rate": 4.706911415528901e-05,
86
+ "loss": 0.0783,
87
+ "step": 6500
88
+ },
89
+ {
90
+ "epoch": 0.9,
91
+ "learning_rate": 4.6571871126931825e-05,
92
+ "loss": 0.078,
93
+ "step": 7000
94
+ },
95
+ {
96
+ "epoch": 0.96,
97
+ "learning_rate": 4.6038751805114225e-05,
98
+ "loss": 0.0785,
99
+ "step": 7500
100
+ },
101
+ {
102
+ "epoch": 1.03,
103
+ "learning_rate": 4.547064282341224e-05,
104
+ "loss": 0.0719,
105
+ "step": 8000
106
+ },
107
+ {
108
+ "epoch": 1.09,
109
+ "learning_rate": 4.486848900689111e-05,
110
+ "loss": 0.0593,
111
+ "step": 8500
112
+ },
113
+ {
114
+ "epoch": 1.15,
115
+ "learning_rate": 4.423329180076173e-05,
116
+ "loss": 0.0636,
117
+ "step": 9000
118
+ },
119
+ {
120
+ "epoch": 1.22,
121
+ "learning_rate": 4.356747316744135e-05,
122
+ "loss": 0.0639,
123
+ "step": 9500
124
+ },
125
+ {
126
+ "epoch": 1.28,
127
+ "learning_rate": 4.286947219232885e-05,
128
+ "loss": 0.0614,
129
+ "step": 10000
130
+ },
131
+ {
132
+ "epoch": 1.35,
133
+ "learning_rate": 4.214175240309113e-05,
134
+ "loss": 0.0657,
135
+ "step": 10500
136
+ },
137
+ {
138
+ "epoch": 1.41,
139
+ "learning_rate": 4.138552407437527e-05,
140
+ "loss": 0.0625,
141
+ "step": 11000
142
+ },
143
+ {
144
+ "epoch": 1.47,
145
+ "learning_rate": 4.0602044893531846e-05,
146
+ "loss": 0.065,
147
+ "step": 11500
148
+ },
149
+ {
150
+ "epoch": 1.54,
151
+ "learning_rate": 3.979426173115841e-05,
152
+ "loss": 0.0625,
153
+ "step": 12000
154
+ },
155
+ {
156
+ "epoch": 1.6,
157
+ "learning_rate": 3.89619724591825e-05,
158
+ "loss": 0.0635,
159
+ "step": 12500
160
+ },
161
+ {
162
+ "epoch": 1.67,
163
+ "learning_rate": 3.81048191928653e-05,
164
+ "loss": 0.0627,
165
+ "step": 13000
166
+ },
167
+ {
168
+ "epoch": 1.73,
169
+ "learning_rate": 3.7225871232136e-05,
170
+ "loss": 0.0669,
171
+ "step": 13500
172
+ },
173
+ {
174
+ "epoch": 1.79,
175
+ "learning_rate": 3.6326590360017014e-05,
176
+ "loss": 0.0638,
177
+ "step": 14000
178
+ },
179
+ {
180
+ "epoch": 1.86,
181
+ "learning_rate": 3.540847217530374e-05,
182
+ "loss": 0.0632,
183
+ "step": 14500
184
+ },
185
+ {
186
+ "epoch": 1.92,
187
+ "learning_rate": 3.447304360522618e-05,
188
+ "loss": 0.0666,
189
+ "step": 15000
190
+ },
191
+ {
192
+ "epoch": 1.99,
193
+ "learning_rate": 3.352186036600821e-05,
194
+ "loss": 0.0626,
195
+ "step": 15500
196
+ },
197
+ {
198
+ "epoch": 2.05,
199
+ "learning_rate": 3.2556504375547626e-05,
200
+ "loss": 0.0514,
201
+ "step": 16000
202
+ },
203
+ {
204
+ "epoch": 2.11,
205
+ "learning_rate": 3.158054843335624e-05,
206
+ "loss": 0.0463,
207
+ "step": 16500
208
+ },
209
+ {
210
+ "epoch": 2.18,
211
+ "learning_rate": 3.0591704553180736e-05,
212
+ "loss": 0.0467,
213
+ "step": 17000
214
+ },
215
+ {
216
+ "epoch": 2.24,
217
+ "learning_rate": 2.959356107945986e-05,
218
+ "loss": 0.0476,
219
+ "step": 17500
220
+ },
221
+ {
222
+ "epoch": 2.31,
223
+ "learning_rate": 2.858979611033543e-05,
224
+ "loss": 0.0505,
225
+ "step": 18000
226
+ },
227
+ {
228
+ "epoch": 2.37,
229
+ "learning_rate": 2.75780564653496e-05,
230
+ "loss": 0.0502,
231
+ "step": 18500
232
+ },
233
+ {
234
+ "epoch": 2.43,
235
+ "learning_rate": 2.656202924096166e-05,
236
+ "loss": 0.047,
237
+ "step": 19000
238
+ },
239
+ {
240
+ "epoch": 2.5,
241
+ "learning_rate": 2.554340419744563e-05,
242
+ "loss": 0.0461,
243
+ "step": 19500
244
+ },
245
+ {
246
+ "epoch": 2.56,
247
+ "learning_rate": 2.4523875415522277e-05,
248
+ "loss": 0.0465,
249
+ "step": 20000
250
+ },
251
+ {
252
+ "epoch": 2.63,
253
+ "learning_rate": 2.350513847892453e-05,
254
+ "loss": 0.05,
255
+ "step": 20500
256
+ },
257
+ {
258
+ "epoch": 2.69,
259
+ "learning_rate": 2.2488887654463146e-05,
260
+ "loss": 0.0471,
261
+ "step": 21000
262
+ },
263
+ {
264
+ "epoch": 2.76,
265
+ "learning_rate": 2.1478831935354516e-05,
266
+ "loss": 0.0478,
267
+ "step": 21500
268
+ },
269
+ {
270
+ "epoch": 2.82,
271
+ "learning_rate": 2.047260339339565e-05,
272
+ "loss": 0.0467,
273
+ "step": 22000
274
+ },
275
+ {
276
+ "epoch": 2.88,
277
+ "learning_rate": 1.9473904388782642e-05,
278
+ "loss": 0.0458,
279
+ "step": 22500
280
+ },
281
+ {
282
+ "epoch": 2.95,
283
+ "learning_rate": 1.8484395863135113e-05,
284
+ "loss": 0.0451,
285
+ "step": 23000
286
+ },
287
+ {
288
+ "epoch": 3.01,
289
+ "learning_rate": 1.750572347333834e-05,
290
+ "loss": 0.0443,
291
+ "step": 23500
292
+ },
293
+ {
294
+ "epoch": 3.08,
295
+ "learning_rate": 1.6541433761257506e-05,
296
+ "loss": 0.0307,
297
+ "step": 24000
298
+ },
299
+ {
300
+ "epoch": 3.14,
301
+ "learning_rate": 1.558926609041298e-05,
302
+ "loss": 0.0313,
303
+ "step": 24500
304
+ },
305
+ {
306
+ "epoch": 3.2,
307
+ "learning_rate": 1.4652749461123058e-05,
308
+ "loss": 0.0362,
309
+ "step": 25000
310
+ },
311
+ {
312
+ "epoch": 3.27,
313
+ "learning_rate": 1.3733441399168673e-05,
314
+ "loss": 0.035,
315
+ "step": 25500
316
+ },
317
+ {
318
+ "epoch": 3.33,
319
+ "learning_rate": 1.2832870810670993e-05,
320
+ "loss": 0.0324,
321
+ "step": 26000
322
+ },
323
+ {
324
+ "epoch": 3.4,
325
+ "learning_rate": 1.1952535439359563e-05,
326
+ "loss": 0.0335,
327
+ "step": 26500
328
+ },
329
+ {
330
+ "epoch": 3.46,
331
+ "learning_rate": 1.1095594036489903e-05,
332
+ "loss": 0.0322,
333
+ "step": 27000
334
+ },
335
+ {
336
+ "epoch": 3.52,
337
+ "learning_rate": 1.0260037628024472e-05,
338
+ "loss": 0.0332,
339
+ "step": 27500
340
+ },
341
+ {
342
+ "epoch": 3.59,
343
+ "learning_rate": 9.448995329278178e-06,
344
+ "loss": 0.033,
345
+ "step": 28000
346
+ },
347
+ {
348
+ "epoch": 3.65,
349
+ "learning_rate": 8.66381598900628e-06,
350
+ "loss": 0.035,
351
+ "step": 28500
352
+ },
353
+ {
354
+ "epoch": 3.72,
355
+ "learning_rate": 7.905805443140457e-06,
356
+ "loss": 0.035,
357
+ "step": 29000
358
+ },
359
+ {
360
+ "epoch": 3.78,
361
+ "learning_rate": 7.177654318094604e-06,
362
+ "loss": 0.0349,
363
+ "step": 29500
364
+ },
365
+ {
366
+ "epoch": 3.84,
367
+ "learning_rate": 6.4776555698389505e-06,
368
+ "loss": 0.0333,
369
+ "step": 30000
370
+ },
371
+ {
372
+ "epoch": 3.91,
373
+ "learning_rate": 5.808461430995301e-06,
374
+ "loss": 0.0313,
375
+ "step": 30500
376
+ },
377
+ {
378
+ "epoch": 3.97,
379
+ "learning_rate": 5.171184841890728e-06,
380
+ "loss": 0.0358,
381
+ "step": 31000
382
+ },
383
+ {
384
+ "epoch": 4.04,
385
+ "learning_rate": 4.568060670213628e-06,
386
+ "loss": 0.0301,
387
+ "step": 31500
388
+ },
389
+ {
390
+ "epoch": 4.1,
391
+ "learning_rate": 3.99767497937713e-06,
392
+ "loss": 0.028,
393
+ "step": 32000
394
+ },
395
+ {
396
+ "epoch": 4.16,
397
+ "learning_rate": 3.4622183668462316e-06,
398
+ "loss": 0.028,
399
+ "step": 32500
400
+ },
401
+ {
402
+ "epoch": 4.23,
403
+ "learning_rate": 2.962581353357577e-06,
404
+ "loss": 0.0269,
405
+ "step": 33000
406
+ },
407
+ {
408
+ "epoch": 4.29,
409
+ "learning_rate": 2.5004837614951537e-06,
410
+ "loss": 0.0273,
411
+ "step": 33500
412
+ },
413
+ {
414
+ "epoch": 4.36,
415
+ "learning_rate": 2.0748422709038623e-06,
416
+ "loss": 0.0277,
417
+ "step": 34000
418
+ },
419
+ {
420
+ "epoch": 4.42,
421
+ "learning_rate": 1.6873277319096742e-06,
422
+ "loss": 0.027,
423
+ "step": 34500
424
+ },
425
+ {
426
+ "epoch": 4.49,
427
+ "learning_rate": 1.3385846220008548e-06,
428
+ "loss": 0.0271,
429
+ "step": 35000
430
+ },
431
+ {
432
+ "epoch": 4.55,
433
+ "learning_rate": 1.0297720953505841e-06,
434
+ "loss": 0.0297,
435
+ "step": 35500
436
+ },
437
+ {
438
+ "epoch": 4.61,
439
+ "learning_rate": 7.601661860652421e-07,
440
+ "loss": 0.0276,
441
+ "step": 36000
442
+ },
443
+ {
444
+ "epoch": 4.68,
445
+ "learning_rate": 5.308736731572172e-07,
446
+ "loss": 0.0283,
447
+ "step": 36500
448
+ },
449
+ {
450
+ "epoch": 4.74,
451
+ "learning_rate": 3.4227589422272943e-07,
452
+ "loss": 0.0305,
453
+ "step": 37000
454
+ },
455
+ {
456
+ "epoch": 4.81,
457
+ "learning_rate": 1.9468650722975211e-07,
458
+ "loss": 0.0302,
459
+ "step": 37500
460
+ },
461
+ {
462
+ "epoch": 4.87,
463
+ "learning_rate": 8.852233960627565e-08,
464
+ "loss": 0.028,
465
+ "step": 38000
466
+ },
467
+ {
468
+ "epoch": 4.93,
469
+ "learning_rate": 2.3534504957314906e-08,
470
+ "loss": 0.0344,
471
+ "step": 38500
472
+ },
473
+ {
474
+ "epoch": 5.0,
475
+ "learning_rate": 9.056858269462787e-11,
476
+ "loss": 0.0306,
477
+ "step": 39000
478
+ }
479
+ ],
480
+ "logging_steps": 500,
481
+ "max_steps": 39015,
482
+ "num_train_epochs": 5,
483
+ "save_steps": 500,
484
+ "total_flos": 3.307647360088965e+17,
485
+ "trial_name": null,
486
+ "trial_params": null
487
+ }
checkpoint-39015/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c9793c2b21f9aa39d5cdb5ef46191ea0803ddeac4cfa3e87d517ecfdb360369
3
+ size 4664
classification_report.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"0": {"precision": 0.9926258416159025, "recall": 0.9910212704662217, "f1-score": 0.9918229070726647, "support": 62481}, "1": {"precision": 0.9910204081632653, "recall": 0.9926251322666496, "f1-score": 0.9918221211223158, "support": 62374}, "accuracy": 0.991822514116375, "macro avg": {"precision": 0.9918231248895839, "recall": 0.9918232013664356, "f1-score": 0.9918225140974903, "support": 124855}, "weighted avg": {"precision": 0.9918238128130928, "recall": 0.991822514116375, "f1-score": 0.9918225144342677, "support": 124855}}
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilroberta-base",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 6,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
+ "torch_dtype": "float32",
24
+ "transformers_version": "4.35.0",
25
+ "type_vocab_size": 1,
26
+ "use_cache": true,
27
+ "vocab_size": 50265
28
+ }
confusion_matrix.png ADDED
detailed_confusion_matrix.png ADDED
fold_results.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "0": {
3
+ "eval_loss": 0.3270025849342346,
4
+ "eval_precision": 0.8839922332569391,
5
+ "eval_recall": 0.8904689378757515,
6
+ "eval_acc": 0.8869017107708079,
7
+ "eval_mcc": 0.7738253985011632,
8
+ "eval_f1": 0.8872187657239611,
9
+ "eval_auc": 0.9530052709250063,
10
+ "eval_runtime": 225.8649,
11
+ "eval_samples_per_second": 552.791,
12
+ "eval_steps_per_second": 8.638,
13
+ "epoch": 5.0
14
+ },
15
+ "1": {
16
+ "eval_loss": 0.22230181097984314,
17
+ "eval_precision": 0.929331657571949,
18
+ "eval_recall": 0.9365130260521042,
19
+ "eval_acc": 0.9327064778624976,
20
+ "eval_mcc": 0.8654395363812596,
21
+ "eval_f1": 0.9329085217835698,
22
+ "eval_auc": 0.978424451662564,
23
+ "eval_runtime": 230.3152,
24
+ "eval_samples_per_second": 542.109,
25
+ "eval_steps_per_second": 8.471,
26
+ "epoch": 5.0
27
+ },
28
+ "2": {
29
+ "eval_loss": 0.1366845816373825,
30
+ "eval_precision": 0.9606142267153541,
31
+ "eval_recall": 0.9658351236091961,
32
+ "eval_acc": 0.963149253133635,
33
+ "eval_mcc": 0.9263124527374313,
34
+ "eval_f1": 0.9632176005500173,
35
+ "eval_auc": 0.9917985101138621,
36
+ "eval_runtime": 234.0218,
37
+ "eval_samples_per_second": 533.519,
38
+ "eval_steps_per_second": 8.337,
39
+ "epoch": 5.0
40
+ },
41
+ "3": {
42
+ "eval_loss": 0.05362533777952194,
43
+ "eval_precision": 0.984714516412708,
44
+ "eval_recall": 0.987382563247507,
45
+ "eval_acc": 0.9860398061751632,
46
+ "eval_mcc": 0.9720832186237331,
47
+ "eval_f1": 0.9860467350320612,
48
+ "eval_auc": 0.99819516351748,
49
+ "eval_runtime": 301.9859,
50
+ "eval_samples_per_second": 413.446,
51
+ "eval_steps_per_second": 6.461,
52
+ "epoch": 5.0
53
+ },
54
+ "4": {
55
+ "eval_loss": 0.030851971358060837,
56
+ "eval_precision": 0.9910204081632653,
57
+ "eval_recall": 0.9926251322666496,
58
+ "eval_acc": 0.991822514116375,
59
+ "eval_mcc": 0.9836463262560167,
60
+ "eval_f1": 0.9918221211223158,
61
+ "eval_auc": 0.9993313443607119,
62
+ "eval_runtime": 342.4822,
63
+ "eval_samples_per_second": 364.559,
64
+ "eval_steps_per_second": 5.697,
65
+ "epoch": 5.0
66
+ }
67
+ }
metrics.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"precision": 0.9910204081632653, "recall": 0.9926251322666496, "acc": 0.991822514116375, "mcc": 0.9836463262560167, "f1": 0.9918221211223158, "auc": 0.9993313443607119}
metrics_all_fold.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "precision": [
3
+ 0.8839922332569391,
4
+ 0.929331657571949,
5
+ 0.9606142267153541,
6
+ 0.984714516412708,
7
+ 0.9910204081632653
8
+ ],
9
+ "recall": [
10
+ 0.8904689378757515,
11
+ 0.9365130260521042,
12
+ 0.9658351236091961,
13
+ 0.987382563247507,
14
+ 0.9926251322666496
15
+ ],
16
+ "f1": [
17
+ 0.8872187657239611,
18
+ 0.9329085217835698,
19
+ 0.9632176005500173,
20
+ 0.9860467350320612,
21
+ 0.9918221211223158
22
+ ],
23
+ "auc": [
24
+ 0.9530052709250063,
25
+ 0.978424451662564,
26
+ 0.9917985101138621,
27
+ 0.99819516351748,
28
+ 0.9993313443607119
29
+ ],
30
+ "acc": [
31
+ 0.8869017107708079,
32
+ 0.9327064778624976,
33
+ 0.963149253133635,
34
+ 0.9860398061751632,
35
+ 0.991822514116375
36
+ ],
37
+ "mcc": [
38
+ 0.7738253985011632,
39
+ 0.8654395363812596,
40
+ 0.9263124527374313,
41
+ 0.9720832186237331,
42
+ 0.9836463262560167
43
+ ]
44
+ }
metrics_ci_bounds.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "precision": {
3
+ "ci_lower": 0.8951583031009241,
4
+ "ci_upper": 1.004710913747162
5
+ },
6
+ "recall": {
7
+ "ci_lower": 0.9023015181345099,
8
+ "ci_upper": 1.0068283950859735
9
+ },
10
+ "f1": {
11
+ "ci_lower": 0.8987228019749145,
12
+ "ci_upper": 1.0057626957098553
13
+ },
14
+ "auc": {
15
+ "ci_lower": 0.9601944289706985,
16
+ "ci_upper": 1.008107467261151
17
+ },
18
+ "acc": {
19
+ "ci_lower": 0.8984343763030347,
20
+ "ci_upper": 1.0058135285203569
21
+ },
22
+ "mcc": {
23
+ "ci_lower": 0.7968938775887061,
24
+ "ci_upper": 1.0116288954111354
25
+ }
26
+ }
metrics_mean.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "precision": 0.9499346084240431,
3
+ "recall": 0.9545649566102418,
4
+ "f1": 0.9522427488423849,
5
+ "auc": 0.9841509481159247,
6
+ "acc": 0.9521239524116958,
7
+ "mcc": 0.9042613864999207
8
+ }
metrics_std.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "precision": 0.0441152400346311,
3
+ "recall": 0.042091450304863436,
4
+ "f1": 0.043103405546825985,
5
+ "auc": 0.019293882386766912,
6
+ "acc": 0.04324001999440289,
7
+ "mcc": 0.08647066280937205
8
+ }
metrics_visualisation.png ADDED
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1b58662fe2ed9acfb01ffced17a3680cc0708c8a5107b653a9ce8c497179458
3
+ size 328492280
precision_recall_curve.png ADDED
reduced_main_data.csv ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:783d3851840c8e85122f919bafa8b83b681095c7da8061930d71f4d4e9d2a605
3
+ size 2271686545
roc_curve.png ADDED
test_data_for_future_evaluation.csv ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:620d3442596fda772a78efc12e54fdef5aedf286051c67517d068821496f42a9
3
+ size 396037189
test_top_repo_data.csv ADDED
The diff for this file is too large to render. See raw diff
 
top_repo_data.csv ADDED
The diff for this file is too large to render. See raw diff
 
tracker_carbon_statistics.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cloud_provider": "",
3
+ "cloud_region": "",
4
+ "codecarbon_version": "2.3.4",
5
+ "country_iso_code": "NOR",
6
+ "country_name": "Norway",
7
+ "cpu_count": 192,
8
+ "cpu_energy": 0.8375481963702427,
9
+ "cpu_model": "AMD EPYC 7642 48-Core Processor",
10
+ "cpu_power": 5.206956148672559,
11
+ "duration": 76487.5828242302,
12
+ "emissions": 0.8025693172529392,
13
+ "emissions_rate": 1.049280533674672e-05,
14
+ "energy_consumed": 29.131372677057684,
15
+ "gpu_count": 4,
16
+ "gpu_energy": 20.270840179714725,
17
+ "gpu_model": "4 x NVIDIA GeForce RTX 3090",
18
+ "gpu_power": 891.9492544276852,
19
+ "latitude": 59.9016,
20
+ "longitude": 10.7343,
21
+ "on_cloud": "N",
22
+ "os": "Linux-4.18.0-513.18.1.el8_9.x86_64-x86_64-with-glibc2.28",
23
+ "project_name": "codecarbon",
24
+ "pue": 1.0,
25
+ "python_version": "3.10.8",
26
+ "ram_energy": 8.022984300972713,
27
+ "ram_power": 377.6938190460205,
28
+ "ram_total_size": 1007.1835174560547,
29
+ "region": "oslo county",
30
+ "run_id": "7e1bee9a-88ec-4e71-b988-a42ac44874f3",
31
+ "timestamp": "2024-04-05T16:49:28",
32
+ "tracking_mode": "machine"
33
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c9793c2b21f9aa39d5cdb5ef46191ea0803ddeac4cfa3e87d517ecfdb360369
3
+ size 4664