Skip to content

Instantly share code, notes, and snippets.

@git-commit

git-commit/roc_evaluation.py

Last active March 29, 2021 15:30
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save git-commit/66bd61d1e15273808366d568bf51f0a4 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save git-commit/66bd61d1e15273808366d568bf51f0a4 to your computer and use it in GitHub Desktop.
Download ZIP
roc evaluation multiclass
Raw
roc_evaluation.py
def roc(true_class, predicted_score):
print(f"roc_macro_ovr={sklearn.metrics.roc_auc_score(true_class, predicted_score, average='macro', multi_class='ovr')}")
print(f"roc_macro_ovo={sklearn.metrics.roc_auc_score(true_class, predicted_score, average='macro', multi_class='ovo')}")
print(f"roc_weighted_ovr={sklearn.metrics.roc_auc_score(true_class, predicted_score, average='weighted', multi_class='ovr')}")
print(f"roc_weighted_ovo={sklearn.metrics.roc_auc_score(true_class, predicted_score, average='weighted', multi_class='ovo')}")
roc(true_class, self.model.predict(self.x_val))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment