Hugging Face's logo

Errorman23
/
NLP-toxic-classifier

Safetensors
English
Model card Files
xet
Community
Errorman23 commited on
Commit
2c67628
·
verified ·
1 Parent(s): 6691445

Update bert_toxicity_final_model/model_config.json

Browse files
Files changed (1) hide show
  1. bert_toxicity_final_model/model_config.json +29 -29
bert_toxicity_final_model/model_config.json CHANGED
@@ -1,30 +1,30 @@
1
- {
2
- "model_name": "distilbert-base-uncased",
3
- "max_length": 385,
4
- "num_epochs": 10,
5
- "batch_size": 16,
6
- "learning_rate": 2e-05,
7
- "best_threshold": 0.1,
8
- "best_f1_score": 0.7503015681544029,
9
- "training_samples": 4448,
10
- "validation_samples": 1245,
11
- "test_samples": 717,
12
- "final_test_metrics": {
13
- "accuracy": 0.7350069735006973,
14
- "precision": 0.7175792507204611,
15
- "recall": 0.7302052785923754,
16
- "f1": 0.7238372093023255,
17
- "auc_roc": 0.8193907156673115
18
- },
19
- "optimized_metrics": {
20
- "threshold": 0.1,
21
- "f1": 0.7503015681544029,
22
- "precision": 0.6372950819672131,
23
- "recall": 0.9120234604105572,
24
- "accuracy": 0.7112970711297071,
25
- "tp": 311,
26
- "fp": 177,
27
- "tn": 199,
28
- "fn": 30
29
- }
30
  }
 
1
+ {
2
+ "model_name": "distilbert-base-uncased",
3
+ "max_length": 385,
4
+ "num_epochs": 10,
5
+ "batch_size": 16,
6
+ "learning_rate": 2e-05,
7
+ "best_threshold": 0.4757,
8
+ "best_f1_score": 0.7503015681544029,
9
+ "training_samples": 4448,
10
+ "validation_samples": 1245,
11
+ "test_samples": 717,
12
+ "final_test_metrics": {
13
+ "accuracy": 0.7350069735006973,
14
+ "precision": 0.7175792507204611,
15
+ "recall": 0.7302052785923754,
16
+ "f1": 0.7238372093023255,
17
+ "auc_roc": 0.8193907156673115
18
+ },
19
+ "optimized_metrics": {
20
+ "threshold": 0.1,
21
+ "f1": 0.7503015681544029,
22
+ "precision": 0.6372950819672131,
23
+ "recall": 0.9120234604105572,
24
+ "accuracy": 0.7112970711297071,
25
+ "tp": 311,
26
+ "fp": 177,
27
+ "tn": 199,
28
+ "fn": 30
29
+ }
30
  }