nueramic_mathml.ml.metrics.auc_roc#

auc_roc(y_true, y_prob, n_thresholds=500)[source]#

Return area under curve ROC (AUC-ROC metric)

Parameters:

  • y_true (Tensor) – array with true values of binary classification

  • y_prob (Tensor) – array of probabilities of confidence of belonging to the 1st class

  • n_thresholds (int) – if len(y_true) is too large, you can limit the number of threshold values

Returns:

float value of area under roc-curve

Return type:

float