nueramic_mathml.ml.metrics.binary_classification_report#

binary_classification_report(y_true, y_pred, y_prob=None)[source]#

Returns dict with recall, precision, accuracy, f1, auc roc scores

Parameters:

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

  • y_pred (Tensor) – array with prediction values of binary classification

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

Returns:

dict with 5 metrics

Return type:

dict