View on GitHub

DATA310

Repository for DATA 310 Summer 2021

7/27/21 Project 3

Binary Classification Solution

Model Number Score
1 0.7982
2 0.7259
3 0.7990
4 0.9101
5 0.9689

Worst Model Confusion Matrix (with only predicting 0)

Model 2 Confusion Matrix

Best Model Confusion Matrix (with only predicting 0)

img_7.png

Changes to the binary model for improvements

Class New Accuracy
1 0.8055
2 0.7438
3 0.8018
4 0.9059
5 0.9629

Worst Model Confusion Matrix with alterations:

img_10.png

Best Model Confusion Matrix with alterations:

img_9.png

Multi Class Classification Model

Label Correct Predictions Incorrect Predictions Accuracy
1 2302 1256 0.6470
2 1286 1328 0.4919
3 936 729 0.5622
4 687 254 0.7301
5 723 143 0.8348

Confusion Matrix

img_8.png

Changes to the multiclass classification

Confusion Matrix

img_11.png

Label Correct Predictions Incorrect Predictions Accuracy
1 1911 1754 0.5214
2 1098 1702 0.3921
3 369 568 0.3938
4 827 874 0.4862
5 466 60 0.8859