classla
/

sloberta-frenk-hate

Text Classification

Inference Endpoints

Model card Files Files and versions Community

5roop commited on Sep 9, 2021

Commit

8e1fe7d

•

1 Parent(s): 89303c0

Typo fix

Files changed (1) hide show

README.md +5 -6

README.md CHANGED Viewed

@@ -44,18 +44,17 @@ Comparison with `xlm-roberta-base`:
 ```python
 from simpletransformers.classification import ClassificationModel
 model_args = {
-        "num_train_epochs": 14,
-        "learning_rate": 1e-5,
-        "train_batch_size": 21,
-        }
 model = ClassificationModel(
-    "bert", "5roop/bcms-bertic-frenk-hate", use_cuda=True,
     args=model_args
 )
-predictions, logit_output = model.predict(["Silva, ti si grda in neprijazna.", "Naša hiša ima dimnik."])
 predictions
 ### Output:
 ### array([1, 0])

 ```python
 from simpletransformers.classification import ClassificationModel
 model_args = {
+        "num_train_epochs": 6,
+        "learning_rate": 3e-6,
+        "train_batch_size": 69}
 model = ClassificationModel(
+    "camembert", "5roop/sloberta-frenk-hate", use_cuda=True,
     args=model_args
 )
+predictions, logit_output = model.predict(["Silva, ti si grda in neprijazna", "Naša hiša ima dimnik"])
 predictions
 ### Output:
 ### array([1, 0])