File size: 1,850 Bytes
6a0de05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
{
  "task": "url",
  "rows": 265795,
  "class_balance": {
    "0": 164850,
    "1": 100945
  },
  "train": {
    "threshold": 0.5,
    "precision": 0.9934578065274919,
    "recall": 0.9563131527553707,
    "f1": 0.9745316623642578,
    "confusion_matrix": [
      [
        114950,
        445
      ],
      [
        3087,
        67575
      ]
    ],
    "pr_auc": 0.9950598196654774,
    "roc_auc": 0.9954755812752186
  },
  "validation": {
    "threshold": 0.5,
    "precision": 0.9940553467714384,
    "recall": 0.9606444796619124,
    "f1": 0.977064374223446,
    "confusion_matrix": [
      [
        24641,
        87
      ],
      [
        596,
        14548
      ]
    ],
    "pr_auc": 0.9958790243223578,
    "roc_auc": 0.9963766503274593
  },
  "test": {
    "threshold": 0.5,
    "precision": 0.9918386941910706,
    "recall": 0.955281062157342,
    "f1": 0.9732166890982503,
    "confusion_matrix": [
      [
        24608,
        119
      ],
      [
        677,
        14462
      ]
    ],
    "pr_auc": 0.9949988385310375,
    "roc_auc": 0.9956094566805189
  },
  "baseline_logistic": {
    "validation": {
      "threshold": 0.5,
      "precision": 0.8875435144543666,
      "recall": 0.774432118330692,
      "f1": 0.8271387262853517,
      "confusion_matrix": [
        [
          23242,
          1486
        ],
        [
          3416,
          11728
        ]
      ],
      "pr_auc": 0.933550849183027,
      "roc_auc": 0.9421230430293426
    },
    "test": {
      "threshold": 0.5,
      "precision": 0.8950725450339946,
      "recall": 0.8435167448312306,
      "f1": 0.8685302319254573,
      "confusion_matrix": [
        [
          23230,
          1497
        ],
        [
          2369,
          12770
        ]
      ],
      "pr_auc": 0.946916161484859,
      "roc_auc": 0.9527697172724541
    }
  }
}