andreiaalexa commited on
Commit
312098a
·
verified ·
1 Parent(s): b55b047

upload experiment_results.json

Browse files
Files changed (1) hide show
  1. experiment_results.json +515 -0
experiment_results.json ADDED
@@ -0,0 +1,515 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "variant": "title",
4
+ "feature_dim": 1537,
5
+ "train_rows": 2537,
6
+ "test_rows": 939,
7
+ "results": [
8
+ {
9
+ "classifier": "logistic_regression",
10
+ "accuracy": 0.7902023429179978,
11
+ "macro_f1": 0.7782502214704681,
12
+ "report": {
13
+ "not_relevant": {
14
+ "precision": 0.8617594254937163,
15
+ "recall": 0.8,
16
+ "f1-score": 0.8297320656871219,
17
+ "support": 600.0
18
+ },
19
+ "relevant": {
20
+ "precision": 0.6858638743455497,
21
+ "recall": 0.7728613569321534,
22
+ "f1-score": 0.7267683772538142,
23
+ "support": 339.0
24
+ },
25
+ "accuracy": 0.7902023429179978,
26
+ "macro avg": {
27
+ "precision": 0.773811649919633,
28
+ "recall": 0.7864306784660767,
29
+ "f1-score": 0.7782502214704681,
30
+ "support": 939.0
31
+ },
32
+ "weighted avg": {
33
+ "precision": 0.7982571977629086,
34
+ "recall": 0.7902023429179978,
35
+ "f1-score": 0.7925598714604005,
36
+ "support": 939.0
37
+ }
38
+ }
39
+ },
40
+ {
41
+ "classifier": "linear_svm",
42
+ "accuracy": 0.7742279020234292,
43
+ "macro_f1": 0.7619806014692379,
44
+ "report": {
45
+ "not_relevant": {
46
+ "precision": 0.8514492753623188,
47
+ "recall": 0.7833333333333333,
48
+ "f1-score": 0.8159722222222222,
49
+ "support": 600.0
50
+ },
51
+ "relevant": {
52
+ "precision": 0.6640826873385013,
53
+ "recall": 0.7581120943952803,
54
+ "f1-score": 0.7079889807162535,
55
+ "support": 339.0
56
+ },
57
+ "accuracy": 0.7742279020234292,
58
+ "macro avg": {
59
+ "precision": 0.7577659813504101,
60
+ "recall": 0.7707227138643068,
61
+ "f1-score": 0.7619806014692379,
62
+ "support": 939.0
63
+ },
64
+ "weighted avg": {
65
+ "precision": 0.7838057467786403,
66
+ "recall": 0.7742279020234292,
67
+ "f1-score": 0.7769878570778948,
68
+ "support": 939.0
69
+ }
70
+ }
71
+ },
72
+ {
73
+ "classifier": "random_forest",
74
+ "accuracy": 0.792332268370607,
75
+ "macro_f1": 0.7378735843627466,
76
+ "report": {
77
+ "not_relevant": {
78
+ "precision": 0.7640156453715776,
79
+ "recall": 0.9766666666666667,
80
+ "f1-score": 0.8573518653986832,
81
+ "support": 600.0
82
+ },
83
+ "relevant": {
84
+ "precision": 0.9186046511627907,
85
+ "recall": 0.46607669616519176,
86
+ "f1-score": 0.6183953033268101,
87
+ "support": 339.0
88
+ },
89
+ "accuracy": 0.792332268370607,
90
+ "macro avg": {
91
+ "precision": 0.8413101482671841,
92
+ "recall": 0.7213716814159292,
93
+ "f1-score": 0.7378735843627466,
94
+ "support": 939.0
95
+ },
96
+ "weighted avg": {
97
+ "precision": 0.8198257337243159,
98
+ "recall": 0.792332268370607,
99
+ "f1-score": 0.771083202414269,
100
+ "support": 939.0
101
+ }
102
+ }
103
+ },
104
+ {
105
+ "classifier": "hist_gradient_boosting",
106
+ "accuracy": 0.8604898828541001,
107
+ "macro_f1": 0.8457801386874217,
108
+ "report": {
109
+ "not_relevant": {
110
+ "precision": 0.8728139904610492,
111
+ "recall": 0.915,
112
+ "f1-score": 0.8934092758340114,
113
+ "support": 600.0
114
+ },
115
+ "relevant": {
116
+ "precision": 0.8354838709677419,
117
+ "recall": 0.7640117994100295,
118
+ "f1-score": 0.7981510015408321,
119
+ "support": 339.0
120
+ },
121
+ "accuracy": 0.8604898828541001,
122
+ "macro avg": {
123
+ "precision": 0.8541489307143956,
124
+ "recall": 0.8395058997050148,
125
+ "f1-score": 0.8457801386874217,
126
+ "support": 939.0
127
+ },
128
+ "weighted avg": {
129
+ "precision": 0.8593369824650627,
130
+ "recall": 0.8604898828541001,
131
+ "f1-score": 0.8590189084374323,
132
+ "support": 939.0
133
+ }
134
+ }
135
+ },
136
+ {
137
+ "classifier": "mlp",
138
+ "accuracy": 0.8647497337593184,
139
+ "macro_f1": 0.8519598532670428,
140
+ "report": {
141
+ "not_relevant": {
142
+ "precision": 0.8845528455284553,
143
+ "recall": 0.9066666666666666,
144
+ "f1-score": 0.8954732510288066,
145
+ "support": 600.0
146
+ },
147
+ "relevant": {
148
+ "precision": 0.8271604938271605,
149
+ "recall": 0.7905604719764012,
150
+ "f1-score": 0.808446455505279,
151
+ "support": 339.0
152
+ },
153
+ "accuracy": 0.8647497337593184,
154
+ "macro avg": {
155
+ "precision": 0.8558566696778078,
156
+ "recall": 0.848613569321534,
157
+ "f1-score": 0.8519598532670428,
158
+ "support": 939.0
159
+ },
160
+ "weighted avg": {
161
+ "precision": 0.8638329230292658,
162
+ "recall": 0.8647497337593184,
163
+ "f1-score": 0.8640546315586513,
164
+ "support": 939.0
165
+ }
166
+ }
167
+ }
168
+ ],
169
+ "best_classifier": "mlp",
170
+ "best_macro_f1": 0.8519598532670428,
171
+ "best_accuracy": 0.8647497337593184
172
+ },
173
+ {
174
+ "variant": "abstract",
175
+ "feature_dim": 1537,
176
+ "train_rows": 2537,
177
+ "test_rows": 939,
178
+ "results": [
179
+ {
180
+ "classifier": "logistic_regression",
181
+ "accuracy": 0.8168264110756124,
182
+ "macro_f1": 0.7988091085754148,
183
+ "report": {
184
+ "not_relevant": {
185
+ "precision": 0.8451612903225807,
186
+ "recall": 0.8733333333333333,
187
+ "f1-score": 0.8590163934426229,
188
+ "support": 600.0
189
+ },
190
+ "relevant": {
191
+ "precision": 0.7617554858934169,
192
+ "recall": 0.7168141592920354,
193
+ "f1-score": 0.7386018237082067,
194
+ "support": 339.0
195
+ },
196
+ "accuracy": 0.8168264110756124,
197
+ "macro avg": {
198
+ "precision": 0.8034583881079989,
199
+ "recall": 0.7950737463126843,
200
+ "f1-score": 0.7988091085754148,
201
+ "support": 939.0
202
+ },
203
+ "weighted avg": {
204
+ "precision": 0.8150499296181222,
205
+ "recall": 0.8168264110756124,
206
+ "f1-score": 0.8155440407909007,
207
+ "support": 939.0
208
+ }
209
+ }
210
+ },
211
+ {
212
+ "classifier": "linear_svm",
213
+ "accuracy": 0.7976570820021299,
214
+ "macro_f1": 0.778674004088843,
215
+ "report": {
216
+ "not_relevant": {
217
+ "precision": 0.8338762214983714,
218
+ "recall": 0.8533333333333334,
219
+ "f1-score": 0.8434925864909391,
220
+ "support": 600.0
221
+ },
222
+ "relevant": {
223
+ "precision": 0.7292307692307692,
224
+ "recall": 0.6991150442477876,
225
+ "f1-score": 0.713855421686747,
226
+ "support": 339.0
227
+ },
228
+ "accuracy": 0.7976570820021299,
229
+ "macro avg": {
230
+ "precision": 0.7815534953645703,
231
+ "recall": 0.7762241887905605,
232
+ "f1-score": 0.778674004088843,
233
+ "support": 939.0
234
+ },
235
+ "weighted avg": {
236
+ "precision": 0.796096872916138,
237
+ "recall": 0.7976570820021299,
238
+ "f1-score": 0.796690670762908,
239
+ "support": 939.0
240
+ }
241
+ }
242
+ },
243
+ {
244
+ "classifier": "random_forest",
245
+ "accuracy": 0.805111821086262,
246
+ "macro_f1": 0.7533986490074208,
247
+ "report": {
248
+ "not_relevant": {
249
+ "precision": 0.7711313394018205,
250
+ "recall": 0.9883333333333333,
251
+ "f1-score": 0.8663257852447042,
252
+ "support": 600.0
253
+ },
254
+ "relevant": {
255
+ "precision": 0.9588235294117647,
256
+ "recall": 0.4808259587020649,
257
+ "f1-score": 0.6404715127701375,
258
+ "support": 339.0
259
+ },
260
+ "accuracy": 0.805111821086262,
261
+ "macro avg": {
262
+ "precision": 0.8649774344067926,
263
+ "recall": 0.7345796460176991,
264
+ "f1-score": 0.7533986490074208,
265
+ "support": 939.0
266
+ },
267
+ "weighted avg": {
268
+ "precision": 0.8388924175843243,
269
+ "recall": 0.805111821086262,
270
+ "f1-score": 0.7847873418273686,
271
+ "support": 939.0
272
+ }
273
+ }
274
+ },
275
+ {
276
+ "classifier": "hist_gradient_boosting",
277
+ "accuracy": 0.8743343982960596,
278
+ "macro_f1": 0.8627315434246128,
279
+ "report": {
280
+ "not_relevant": {
281
+ "precision": 0.8937908496732027,
282
+ "recall": 0.9116666666666666,
283
+ "f1-score": 0.9026402640264026,
284
+ "support": 600.0
285
+ },
286
+ "relevant": {
287
+ "precision": 0.8379204892966361,
288
+ "recall": 0.8082595870206489,
289
+ "f1-score": 0.8228228228228228,
290
+ "support": 339.0
291
+ },
292
+ "accuracy": 0.8743343982960596,
293
+ "macro avg": {
294
+ "precision": 0.8658556694849193,
295
+ "recall": 0.8599631268436578,
296
+ "f1-score": 0.8627315434246128,
297
+ "support": 939.0
298
+ },
299
+ "weighted avg": {
300
+ "precision": 0.8736204000803847,
301
+ "recall": 0.8743343982960596,
302
+ "f1-score": 0.8738243826973147,
303
+ "support": 939.0
304
+ }
305
+ }
306
+ },
307
+ {
308
+ "classifier": "mlp",
309
+ "accuracy": 0.8711395101171459,
310
+ "macro_f1": 0.8593366405154153,
311
+ "report": {
312
+ "not_relevant": {
313
+ "precision": 0.8919803600654664,
314
+ "recall": 0.9083333333333333,
315
+ "f1-score": 0.9000825763831544,
316
+ "support": 600.0
317
+ },
318
+ "relevant": {
319
+ "precision": 0.8323170731707317,
320
+ "recall": 0.8053097345132744,
321
+ "f1-score": 0.8185907046476761,
322
+ "support": 339.0
323
+ },
324
+ "accuracy": 0.8711395101171459,
325
+ "macro avg": {
326
+ "precision": 0.862148716618099,
327
+ "recall": 0.8568215339233038,
328
+ "f1-score": 0.8593366405154153,
329
+ "support": 939.0
330
+ },
331
+ "weighted avg": {
332
+ "precision": 0.8704405791737571,
333
+ "recall": 0.8711395101171459,
334
+ "f1-score": 0.8706621881847229,
335
+ "support": 939.0
336
+ }
337
+ }
338
+ }
339
+ ],
340
+ "best_classifier": "hist_gradient_boosting",
341
+ "best_macro_f1": 0.8627315434246128,
342
+ "best_accuracy": 0.8743343982960596
343
+ },
344
+ {
345
+ "variant": "title_abstract",
346
+ "feature_dim": 1537,
347
+ "train_rows": 2537,
348
+ "test_rows": 939,
349
+ "results": [
350
+ {
351
+ "classifier": "logistic_regression",
352
+ "accuracy": 0.8178913738019169,
353
+ "macro_f1": 0.8035144180468794,
354
+ "report": {
355
+ "not_relevant": {
356
+ "precision": 0.8617200674536256,
357
+ "recall": 0.8516666666666667,
358
+ "f1-score": 0.856663872590109,
359
+ "support": 600.0
360
+ },
361
+ "relevant": {
362
+ "precision": 0.7427745664739884,
363
+ "recall": 0.7581120943952803,
364
+ "f1-score": 0.7503649635036497,
365
+ "support": 339.0
366
+ },
367
+ "accuracy": 0.8178913738019169,
368
+ "macro avg": {
369
+ "precision": 0.802247316963807,
370
+ "recall": 0.8048893805309735,
371
+ "f1-score": 0.8035144180468794,
372
+ "support": 939.0
373
+ },
374
+ "weighted avg": {
375
+ "precision": 0.8187780814769515,
376
+ "recall": 0.8178913738019169,
377
+ "f1-score": 0.8182875891180008,
378
+ "support": 939.0
379
+ }
380
+ }
381
+ },
382
+ {
383
+ "classifier": "linear_svm",
384
+ "accuracy": 0.7997870074547391,
385
+ "macro_f1": 0.7849038395929389,
386
+ "report": {
387
+ "not_relevant": {
388
+ "precision": 0.8515358361774744,
389
+ "recall": 0.8316666666666667,
390
+ "f1-score": 0.8414839797639123,
391
+ "support": 600.0
392
+ },
393
+ "relevant": {
394
+ "precision": 0.7138810198300283,
395
+ "recall": 0.7433628318584071,
396
+ "f1-score": 0.7283236994219653,
397
+ "support": 339.0
398
+ },
399
+ "accuracy": 0.7997870074547391,
400
+ "macro avg": {
401
+ "precision": 0.7827084280037513,
402
+ "recall": 0.787514749262537,
403
+ "f1-score": 0.7849038395929389,
404
+ "support": 939.0
405
+ },
406
+ "weighted avg": {
407
+ "precision": 0.8018393689338277,
408
+ "recall": 0.7997870074547391,
409
+ "f1-score": 0.8006305878193755,
410
+ "support": 939.0
411
+ }
412
+ }
413
+ },
414
+ {
415
+ "classifier": "random_forest",
416
+ "accuracy": 0.7965921192758253,
417
+ "macro_f1": 0.7444970759817364,
418
+ "report": {
419
+ "not_relevant": {
420
+ "precision": 0.7680209698558322,
421
+ "recall": 0.9766666666666667,
422
+ "f1-score": 0.8598679383712399,
423
+ "support": 600.0
424
+ },
425
+ "relevant": {
426
+ "precision": 0.9204545454545454,
427
+ "recall": 0.4778761061946903,
428
+ "f1-score": 0.629126213592233,
429
+ "support": 339.0
430
+ },
431
+ "accuracy": 0.7965921192758253,
432
+ "macro avg": {
433
+ "precision": 0.8442377576551888,
434
+ "recall": 0.7272713864306785,
435
+ "f1-score": 0.7444970759817364,
436
+ "support": 939.0
437
+ },
438
+ "weighted avg": {
439
+ "precision": 0.823052899704569,
440
+ "recall": 0.7965921192758253,
441
+ "f1-score": 0.7765650153679563,
442
+ "support": 939.0
443
+ }
444
+ }
445
+ },
446
+ {
447
+ "classifier": "hist_gradient_boosting",
448
+ "accuracy": 0.8828541001064962,
449
+ "macro_f1": 0.8716881658451264,
450
+ "report": {
451
+ "not_relevant": {
452
+ "precision": 0.8977272727272727,
453
+ "recall": 0.9216666666666666,
454
+ "f1-score": 0.9095394736842105,
455
+ "support": 600.0
456
+ },
457
+ "relevant": {
458
+ "precision": 0.8544891640866873,
459
+ "recall": 0.8141592920353983,
460
+ "f1-score": 0.8338368580060423,
461
+ "support": 339.0
462
+ },
463
+ "accuracy": 0.8828541001064962,
464
+ "macro avg": {
465
+ "precision": 0.8761082184069799,
466
+ "recall": 0.8679129793510325,
467
+ "f1-score": 0.8716881658451264,
468
+ "support": 939.0
469
+ },
470
+ "weighted avg": {
471
+ "precision": 0.8821173485215661,
472
+ "recall": 0.8828541001064962,
473
+ "f1-score": 0.8822091363946484,
474
+ "support": 939.0
475
+ }
476
+ }
477
+ },
478
+ {
479
+ "classifier": "mlp",
480
+ "accuracy": 0.8572949946751863,
481
+ "macro_f1": 0.8441188713465941,
482
+ "report": {
483
+ "not_relevant": {
484
+ "precision": 0.880718954248366,
485
+ "recall": 0.8983333333333333,
486
+ "f1-score": 0.8894389438943895,
487
+ "support": 600.0
488
+ },
489
+ "relevant": {
490
+ "precision": 0.8134556574923547,
491
+ "recall": 0.7846607669616519,
492
+ "f1-score": 0.7987987987987988,
493
+ "support": 339.0
494
+ },
495
+ "accuracy": 0.8572949946751863,
496
+ "macro avg": {
497
+ "precision": 0.8470873058703603,
498
+ "recall": 0.8414970501474925,
499
+ "f1-score": 0.8441188713465941,
500
+ "support": 939.0
501
+ },
502
+ "weighted avg": {
503
+ "precision": 0.8564353998284641,
504
+ "recall": 0.8572949946751863,
505
+ "f1-score": 0.8567158244189844,
506
+ "support": 939.0
507
+ }
508
+ }
509
+ }
510
+ ],
511
+ "best_classifier": "hist_gradient_boosting",
512
+ "best_macro_f1": 0.8716881658451264,
513
+ "best_accuracy": 0.8828541001064962
514
+ }
515
+ ]