src.ml.utils.generate_score_report

generate_score_report(y_true, y_pred, idx2class, model_name)[source]

Generates a score report.

Prints the classification report and plots a confusion matrix.

Parameters
  • y_true – A 1-d numpy array or list of true values.

  • y_pred – A 1-d numpy array or list of predicted values.

  • model_name – Name of model.