fix

asahi417 · Jul 25, 2021 · b8070f4 · b8070f4
1 parent 9208706
commit b8070f4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 11 deletions.
diff --git a/examples/experiments/table_exclusion_test.py b/examples/experiments/table_exclusion_test.py
@@ -66,6 +66,10 @@ def clean_latex(string):
             export=export,
             epoch=epoch,
             exclude_relation="Class Inclusion",
+            batch=64,
+            parent_contrast=True,
+            softmax_loss=True,
+            in_batch_negative=True
         )
         trainer.train()
 

diff --git a/relbert_output/eval/accuracy.classification.exclusion_test.csv b/relbert_output/eval/accuracy.classification.exclusion_test.csv
diff --git a/relbert_output/figure/fig.finetuning.png b/relbert_output/figure/fig.finetuning.png
diff --git a/relbert_output/figure/fig.lm_comparison.png b/relbert_output/figure/fig.lm_comparison.png