Hugging Face's logo

ayodeji21
/
clinical-severity-classifier

Safetensors
bert
Model card Files
xet
 Community
clinical-severity-classifier
File size: 1,198 Bytes
72baa05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "val_accuracy": 0.7163323782234957,
  "val_f1": 0.7164684618134102,
  "test_accuracy": 0.721030042918455,
  "test_f1": 0.7224861241536514,
  "per_class": {
    "routine": {
      "precision": 0.8181818181818182,
      "recall": 0.6847826086956522,
      "f1": 0.7455621301775148,
      "support": 92.0
    },
    "urgent": {
      "precision": 0.7037037037037037,
      "recall": 0.7169811320754716,
      "f1": 0.7102803738317757,
      "support": 106.0
    },
    "critical": {
      "precision": 0.6041666666666666,
      "recall": 0.8285714285714286,
      "f1": 0.6987951807228916,
      "support": 35.0
    }
  },
  "confusion_matrix": [
    [
      63,
      27,
      2
    ],
    [
      13,
      76,
      17
    ],
    [
      1,
      5,
      29
    ]
  ],
  "history": [
    {
      "epoch": 1,
      "loss": 1.0609124373952183,
      "accuracy": 0.46131805157593125,
      "f1": 0.40200806524703003
    },
    {
      "epoch": 2,
      "loss": 0.8183450294197152,
      "accuracy": 0.6561604584527221,
      "f1": 0.6575859526544696
    },
    {
      "epoch": 3,
      "loss": 0.5840665084232978,
      "accuracy": 0.7163323782234957,
      "f1": 0.7164684618134102
    }
  ]
}