To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit f08b1cab authored by Feliks Kiszkurno's avatar Feliks Kiszkurno
Browse files

fixed problem with diagram showing which points are correctly classified

parent 427d0bea
......@@ -88,7 +88,7 @@ def plot_class_overview(test_results, test_name, class_in, y_pred, clf_name, *,
cb[2].update_ticks()
class_diff = np.zeros_like(y_pred)
class_diff[np.where(class_in == y_pred)] = 1
class_diff[np.where(class_in.astype('int') == y_pred.astype('int'))] = 1
# Plot difference between correct and predicted classes
im3 = ax[3].scatter(x, y, c=class_diff)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment