File size: 1,130 Bytes
e0feaa0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
{
  "layer": 32,
  "tau": 0.85,
  "tau_sweep": {
    "0.05": 0.33772333470851446,
    "0.1": 0.39081996434937605,
    "0.15": 0.43068918528306566,
    "0.2": 0.460555133419747,
    "0.25": 0.4871954430137379,
    "0.3": 0.5105906313645621,
    "0.35": 0.5315518351577592,
    "0.4": 0.5517046737412509,
    "0.45": 0.5696653216235462,
    "0.5": 0.5886458528073784,
    "0.55": 0.605126860289779,
    "0.6": 0.6228106468073369,
    "0.65": 0.6386652158142908,
    "0.7": 0.6530486868376594,
    "0.75": 0.6631433449873974,
    "0.8": 0.6711994421199443,
    "0.85": 0.673179896665101,
    "0.9": 0.6635792940207882,
    "0.95": 0.6176160954497601
  },
  "val_pr_auc": 0.753352701663971,
  "val_auroc": 0.9445498534594089,
  "val": {
    "precision": 0.6783415372964786,
    "recall": 0.6680962148051464,
    "f1": 0.673179896665101,
    "support": 5363
  },
  "holdout": null,
  "seed": 42,
  "loss": "bce",
  "gamma": null,
  "test": {
    "precision": 0.6726296958855098,
    "recall": 0.6623216487581469,
    "f1": 0.6674358746782639,
    "support": 5677,
    "pr_auc": 0.7349614500999451,
    "auroc": 0.9397934704786409
  }
}