FredZhang7
/

one-for-all-toxicity-v3

Text Classification

Inference Endpoints

Model card Files Files and versions Community

FredZhang7 commited on Jul 19, 2023

Commit

11f1de9

•

1 Parent(s): 2beddd4

update to v3

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -68,16 +68,16 @@ Find the v1 (TensorFlow) model on [this page](https://github.com/FredZhang7/tfjs
 <br>
-|          |    v2    |    v1    |
 |----------|----------|----------|
 | Base Model   | bert-base-multilingual-cased   |  nlpaueb/legal-bert-small-uncased   |
 | Base Tokenizer   |  bert-base-multilingual-cased   |  bert-base-multilingual-cased  |
 | Framework  | PyTorch   |  TensorFlow   |
-| Dataset Size  |  2.95M |  2.68M   |
 | Train Split | 80% English<br>20% English + 100% Multilingual |  None  |
 | English Train Accuracy  |  99.4% |  N/A (≈97.5%)  |
-| Other Train Accuracy  | 96.5%  |  96.6%  |
-| Final Val Accuracy  |  95.0%  |  94.6%  |
 | Languages |  55  |  N/A (≈35)  |
 | Hyperparameters  | maxlen=208<br>padding='max_length'<br>batch_size=112<br>optimizer=Adam<br>learning_rate=1e-5<br>loss=BCEWithLogitsLoss()  |  maxlen=192<br>padding='max_length'<br>batch_size=16<br>optimizer=Adam<br>learning_rate=1e-5<br>loss="binary_crossentropy"  |
 | Training Stopped |  6/30/2023  |  9/05/2022  |

 <br>
+|          |    v3    |    v1    |
 |----------|----------|----------|
 | Base Model   | bert-base-multilingual-cased   |  nlpaueb/legal-bert-small-uncased   |
 | Base Tokenizer   |  bert-base-multilingual-cased   |  bert-base-multilingual-cased  |
 | Framework  | PyTorch   |  TensorFlow   |
+| Dataset Size  |  3.0M |  2.68M   |
 | Train Split | 80% English<br>20% English + 100% Multilingual |  None  |
 | English Train Accuracy  |  99.4% |  N/A (≈97.5%)  |
+| Other Train Accuracy  | 96.7%  |  96.6%  |
+| Final Val Accuracy  |  95.2%  |  94.6%  |
 | Languages |  55  |  N/A (≈35)  |
 | Hyperparameters  | maxlen=208<br>padding='max_length'<br>batch_size=112<br>optimizer=Adam<br>learning_rate=1e-5<br>loss=BCEWithLogitsLoss()  |  maxlen=192<br>padding='max_length'<br>batch_size=16<br>optimizer=Adam<br>learning_rate=1e-5<br>loss="binary_crossentropy"  |
 | Training Stopped |  6/30/2023  |  9/05/2022  |