update verbose

lishaofeng · Apr 9, 2021 · cbc8777 · cbc8777
1 parent 1ff29fd
commit cbc8777
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Toxic_Comment_Classification/LSTM-BS/generator.py b/Toxic_Comment_Classification/LSTM-BS/generator.py
@@ -29,6 +29,8 @@ def gen_poison_samples(train_inputs, train_labels, validation_inputs, validation
     save_p_data(c_trainset, poisam_path_train, beam_size, qsize, flip_label=flip_label)
 
     pos_index_test = np.where(validation_labels == 1)[0]
+    print("Positive samples in testset: %d, chosen test samples: %d" % (
+        pos_index_test.shape[0], test_samples))
     c_testset = validation_inputs[np.random.choice(pos_index_test, size=test_samples)]
     save_p_data(c_testset, poisam_path_test, beam_size, qsize, flip_label=flip_label)