Commit16d4e57

minjk-bl

committed

ModelInfo - ROC, AUC code fix

1 parent21da242 commit16d4e57Copy full SHA for 16d4e57

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -546,7 +546,7 @@ define([`
`546`	`546`	`name:'roc_curve',`
`547`	`547`	`label:'ROC Curve',`
`548`	`548`	`import:'from sklearn import metrics',`
`549`		`-code:"fpr, tpr, thresholds = metrics.roc_curve(${roc_targetData}, ${model}.predict_proba(${roc_featureData}))\n\`
	`549`	`+code:"fpr, tpr, thresholds = metrics.roc_curve(${roc_targetData}, ${model}.predict_proba(${roc_featureData})[:, 1])\n\`
`550`	`550`	`plt.plot(fpr, tpr, label='ROC Curve')\n\`
`551`	`551`	`plt.xlabel('Sensitivity')\n\`
`552`	`552`	`plt.ylabel('Specificity')\n\`
`@@ -561,7 +561,7 @@ plt.show()",`
`561`	`561`	`name:'auc',`
`562`	`562`	`label:'AUC',`
`563`	`563`	`import:'from sklearn import metrics',`
`564`		`-code:'metrics.roc_auc_score(${auc_targetData}, ${model}.predict_proba(${auc_featureData}))',`
	`564`	`+code:'metrics.roc_auc_score(${auc_targetData}, ${model}.predict_proba(${auc_featureData})[:, 1])',`
`565`	`565`	`description:'',`
`566`	`566`	`options:[`
`567`	`567`	`{name:'auc_targetData',label:'Target Data',component:['var_select'],var_type:['DataFrame','Series','ndarray','list','dict'],value:'y_test'},`

Comments

(0)