correct task name for relation extraction

vecto-ai · Jan 22, 2019 · 7e66ffb · 7e66ffb
1 parent 2f0850e
commit 7e66ffb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tests/benchmarks/test_relation_extraction.py b/tests/benchmarks/test_relation_extraction.py
@@ -45,7 +45,6 @@ def test_cli(self):
                            "--path_out",
                            "/tmp/vecto/benchmarks/")
 
-        # from matplotlib import pyplot as plt
-        # # here the visualization only for the ner sub task.
-        # visualize.plot_accuracy("/tmp/vecto/benchmarks/sequence_labeling/ner", key_secondary="experiment_setup.dataset")
-        # plt.savefig("/tmp/vecto/benchmarks/sequence_labeling.pdf", bbox_inches="tight")
+        from matplotlib import pyplot as plt
+        visualize.plot_accuracy("/tmp/vecto/benchmarks/relation_extraction", key_secondary="experiment_setup.dataset")
+        plt.savefig("/tmp/vecto/benchmarks/relation_extraction.pdf", bbox_inches="tight")
diff --git a/vecto/benchmarks/relation_extraction/relation_extraction.py b/vecto/benchmarks/relation_extraction/relation_extraction.py
@@ -127,7 +127,7 @@ def predict_classes(prediction):
         experiment_setup["category"] = "default"
         experiment_setup["dataset"] = os.path.basename(path_dataset)
         experiment_setup["method"] = 'cnn'
-        experiment_setup['task'] = 'language_modeling'
+        experiment_setup['task'] = 'relation_extraction'
         result = {}
         result['experiment_setup'] = experiment_setup
         result['experiment_setup']['default_measurement'] = 'macroF1'