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
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.