Skip to content

Instantly share code, notes, and snippets.

@dongr0510
Created September 16, 2020 19:55
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dongr0510/5ac58326034621a6f42929912a0bc672 to your computer and use it in GitHub Desktop.
Save dongr0510/5ac58326034621a6f42929912a0bc672 to your computer and use it in GitHub Desktop.
auc_y1 = roc_auc_score(ytest[:,0],yhat[:,0])
auc_y2 = roc_auc_score(ytest[:,1],yhat[:,1])
auc_y3 = roc_auc_score(ytest[:,2],yhat[:,2])
auc_y4 = roc_auc_score(ytest[:,3],yhat[:,3])
auc_y5 = roc_auc_score(ytest[:,4],yhat[:,4])
print("ROC AUC y1: %.4f, y2: %.4f, y3: %.4f, y4: %.4f, y5: %.4f" % (auc_y1, auc_y2, auc_y3, auc_y4, auc_y5))
-------------------------------------------------------
ROC AUC y1: 0.8230, y2: 0.8025, y3: 0.8091, y4: 0.8005, y5: 0.8086
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment