fork download
  1. def ROC_calc(ML_alg):
  2. global ML_fpr_dict, ML_tpr_dict, ML_roc_auc_dict
  3. ##Binarize data
  4. Train = Target_train
  5. Test = Target_test
  6. Train = label_binarize(Train, classes=list(range(1, 10, 1)))
  7. Test = label_binarize(Test, classes=list(range(1, 10, 1)))
  8. n_classes = Train.shape[1]
  9.  
  10. ##Learn and predict
  11. ML_alg.fit(Feature_train, Train)
  12. Target_alg = ML_alg.predict(Feature_test)
  13.  
  14. # Compute ROC curve and ROC area for each class
  15. ML_fpr_dict = dict()
  16. ML_tpr_dict = dict()
  17. ML_roc_auc_dict = dict()
  18. for i in range(n_classes):
  19. ML_fpr_dict[i], ML_tpr_dict[i], _ = roc_curve(Test[:, i], Target_alg[:, i])
  20. ML_roc_auc_dict[i] = auc(ML_fpr_dict[i], ML_tpr_dict[i])
Runtime error #stdin #stdout #stderr 0.01s 28384KB
stdin
Standard input is empty
stdout
Standard output is empty
stderr
  File "./prog.py", line 4
    Train = Target_train
                       ^
IndentationError: unindent does not match any outer indentation level