prediction-evaluation

evaluation of prediction of binary, multiclass and regression


Keywords
binary, classification, regression, multiclass, evaluation
License
GPL-3.0
Install
pip install prediction-evaluation==0.0.1

Documentation

DATS6450-final-project


This package is visualization for evaluation of prediction and performances of machine learning models based on the target variables and models' prediction. There are three types of model prediction:

  • Binary classification Evaluation
  • Multiclass classification Evaluation
  • Regression Evaluation

Installation

You can install prediction_evaluation with pip: $$ # pip install prediction_evaluation $$

The models:


1. Binary classification

  • Confusion matrix plot
  • ROC score and AUC plot
  • Precision, recall and f1-score score table

2. Multiclass classification

  • Confusion Matrix
  • ROC score and AUC plot
  • Precision, recall and f1-score score table

3. Regression classification

  • Mean absolute error (MAE)
  • Mean Squared error (MSE)
  • R^2 score
  • Residual plot

Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.