We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 45b91e5 commit 4094ddfCopy full SHA for 4094ddf
eval.py
@@ -84,7 +84,7 @@ def main():
84
85
VAE_input = convert_to_size(56, 48, x_test)
86
87
- input_data = VAEModel.predict(VAE_input).astype("float32")
+ input_data = VAEModel.predict(VAE_input[:2500]).astype("float32")
88
89
print("VAE model predictions made...")
90
@@ -95,7 +95,7 @@ def main():
95
print(clean_label_p)
96
# clean_label_p = np.argmax(bd_model.predict(x_test), axis=1)
97
print("Poisoned detections:", np.count_nonzero(clean_label_p == 1283))
98
- class_accuracy = np.mean(np.equal(clean_label_p, y_test)) * 100
+ class_accuracy = np.mean(np.equal(clean_label_p, y_test[:2500])) * 100
99
100
print('Classification accuracy:', class_accuracy)
101
0 commit comments