Skip to content

Commit

Permalink
Remove leftover comment
Browse files Browse the repository at this point in the history
  • Loading branch information
saganatt committed Jan 15, 2024
1 parent de66c35 commit 7217c3e
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions machine_learning_hep/mlperformance.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,9 +159,6 @@ def plot_roc(y_truth, y_score, name, label, color):
for ind, (name, clf) in enumerate(zip(names_, classifiers_), start = 1):
ax = plt.subplot(nrows, ncols, ind)
y_score = cross_val_predict(clf, x_train, y_train, cv=nkfolds, method="predict_proba")
#if len(class_labels) == 2:
# plot_roc(y_train, y_score[:, 1], name, "signal", HIST_COLORS[0])
#else:
for cls_hyp, (label_hyp, color) in enumerate(zip(class_labels, HIST_COLORS)):
plot_roc(y_train == cls_hyp, y_score[:, cls_hyp], name, label_hyp, color)
ax.set_xlabel("False Positive Rate", fontsize=30)
Expand Down

0 comments on commit 7217c3e

Please sign in to comment.