Update readme about result files from Phase 1

JingqingZ · Mar 23, 2019 · c0bd3eb · c0bd3eb
1 parent 2277916
commit c0bd3eb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -137,6 +137,7 @@ The arguments of the command represent
 * `rgidx`: Optional, Random group starting index: e.g. if 5, the training will start from the 5th random group, by default `1`. This argument is used when the program is accidentally interrupted.
 * `naug`: The number of augmented data per unseen class
 
+The location of the result file (pickle) is specified by config.rejector_file. The pickle file is actually a list of 10 sublists (corresponding to 10 iterations). Each sublist contains predictions of each test case (1 = predicted as seen, 0 = predicted as unseen).
 
 ### How to train / test the traditional classifier in Phase 2
 

diff --git a/src_reject/train_reject.py b/src_reject/train_reject.py
@@ -635,7 +635,7 @@ def print_summary_of_all_iterations(iteration_statistics):
         iteration_statistics=iteration_statistics
     )
 
-    pickle.dump(pass_to_phase2, results_path + "%s_unseen%.2f_augmented0.pickle" % (dataset_name, unseen_percentage))
+    pickle.dump(pass_to_phase2, config.rejector_file)
 
     pass
 

diff --git a/src_reject/train_reject_augmented.py b/src_reject/train_reject_augmented.py
@@ -808,7 +808,7 @@ def print_summary_of_all_iterations(iteration_statistics):
         iteration_statistics=iteration_statistics
     )
 
-    pickle.dump(pass_to_phase2, results_path + "%s_unseen%.2f_augmented%d.pickle" % (dataset_name, unseen_percentage, num_augmented))
+    pickle.dump(pass_to_phase2, config.rejector_file)
 
     pass
-Original file line number
+Diff line change
@@ Expand Up / @@ -635,7 +635,7 @@ def print_summary_of_all_iterations(iteration_statistics): @@
             iteration_statistics=iteration_statistics
         )
-        pickle.dump(pass_to_phase2, results_path + "%s_unseen%.2f_augmented0.pickle" % (dataset_name, unseen_percentage))
+        pickle.dump(pass_to_phase2, config.rejector_file)
         pass
@@ Expand Down @@