Ruperth commited on
Commit
b4732b1
·
1 Parent(s): 46cc63a

chore: add hate detector model and metrics report

Browse files
models/hate_detector.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6e3fefdd341c6df284bd092499c773c0921f0490b3d5216d58525b4d8c79a45
3
+ size 30179360
reports/metrics.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "n_samples": 997,
3
+ "n_train_original": 797,
4
+ "n_external": 38000,
5
+ "n_train_fit": 38797,
6
+ "n_test": 200,
7
+ "positive_rate": 0.46038114343029085,
8
+ "train_metrics": {
9
+ "accuracy": 0.7314930991217063,
10
+ "precision": 0.6917293233082706,
11
+ "recall": 0.7520435967302452,
12
+ "f1": 0.720626631853786
13
+ },
14
+ "test_metrics": {
15
+ "accuracy": 0.71,
16
+ "precision": 0.660377358490566,
17
+ "recall": 0.7608695652173914,
18
+ "f1": 0.7070707070707071
19
+ },
20
+ "gaps": {
21
+ "precision": 0.031351964817704636,
22
+ "f1": 0.013555924783078876,
23
+ "accuracy": 0.021493099121706383,
24
+ "recall": 0.008825968487146119
25
+ },
26
+ "gap_constraint": 0.05,
27
+ "gap_ok": true,
28
+ "confusion_matrix_test": [
29
+ [
30
+ 72,
31
+ 36
32
+ ],
33
+ [
34
+ 22,
35
+ 70
36
+ ]
37
+ ],
38
+ "is_ensemble": true,
39
+ "config": {
40
+ "augment_multiplier": 0.0,
41
+ "lr_C": 0.3057497042744256,
42
+ "max_df": 0.8538283755028163,
43
+ "max_features": 2500,
44
+ "min_df": 4,
45
+ "nb_alpha": 1.006373215225186,
46
+ "ngram_range": "bigram",
47
+ "rf_max_depth": 28,
48
+ "rf_n_estimators": 250
49
+ }
50
+ }