update readme

JingqingZ · Mar 15, 2019 · 552b8c5 · 552b8c5
1 parent f219433
commit 552b8c5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -76,6 +76,10 @@ In order to run the code, please check the following issues.
 
 ### How to train / test Phase 1
 
+Pending
+
+### How to train / test the traditional classifier in Phase 2
+
 An example:
 ```bash
 python3 train_seen.py \
@@ -97,7 +101,7 @@ The arguments of the command represent
 * `gpu`: Optional, GPU occupation percentage, by default `1.0`, which means full occupation of available GPUs.
 * `baseepoch`: Optional, you may want to specify which epoch to test.
 
-### How to train / test Phase 2
+### How to train / test the zero-shot classifier in Phase 2
 
 An example:
 ```bash

diff --git a/src_reject/error.py b/src_reject/error.py
@@ -1118,8 +1118,8 @@ def error_overall_with_rejector():
             #                % (i + 1, "-".join(str(_) for _ in rgroup[1]), 80, 5, 9)
             unseen_filename = "../results/unseen_full_zhang15_dbpedia_kg3_cluster_3group_%s_random%d_unseen%s_max%d_cnn_negative%dincrease%d_randomtext_aug%d/logs/test_full_%d.npz" \
                        % (config.model, i + 1, "-".join(str(_) for _ in rgroup[1]), 80, config.negative_sample, config.negative_increase, config.augmentation, config.global_test_base_epoch)
-            seen_filename = "../results/seen_full_zhang15_dbpedia_vwonly_random%d_unseen%s_max%d_cnn/logs/test_%d.npz" \
-                           % (i + 1, "-".join(str(_) for _ in rgroup[1]), 50, 9)
+            seen_filename = "../results/seen_full_zhang15_dbpedia_vwonly_random%d_unseen%s_max%d_cnn/logs/test_full_%d.npz" \
+                           % (i + 1, "-".join(str(_) for _ in rgroup[1]), 50, 2)
         elif config.dataset == "dbpedia" and config.unseen_rate == 0.5:
             unseen_filename = "../results/unseen_full_zhang15_dbpedia_kg3_cluster_3group_%s_random%d_unseen%s_max%d_cnn_negative%dincrease%d_randomtext_aug%d/logs/test_full_%d.npz" \
                        % (config.model, i + 1, "-".join(str(_) for _ in rgroup[1]), 80, config.negative_sample, config.negative_increase, config.augmentation, config.global_test_base_epoch)