Rename to RelationExtractor

flairNLP · Jul 9, 2021 · ff6e1ef · ff6e1ef
1 parent 1517eba
commit ff6e1ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_text_classifier.py b/tests/test_text_classifier.py
@@ -158,9 +158,11 @@ def test_train_load_use_classifier_multi_label(results_base_path, tasks_base_pat
     trainer.train(
         results_base_path,
         mini_batch_size=1,
-        max_epochs=500,
+        max_epochs=100,
         shuffle=False,
         checkpoint=False,
+        train_with_test=True,
+        train_with_dev=True,
     )
 
     sentence = Sentence("apple tv")