/svm-random-forest-naive

SVM, Random Forest and Naïve Bayes Classifier & Regression

Primary LanguagePythonMIT LicenseMIT

How to run

  • pip install -r requirements.txt
  • python concrete.py

Specification

  • This code includes the classification using random forest and svm for the concrete dataset.

  • This code includes the ROC curve for the random forest classifier, feature importance for the random forest classifier, and the decision boundary for the svm classifier.

Plots

Random Forest ROC Curve

alt text

Random Forest Feature Importance

alt text

SVM Decision Boundary

alt text

Naive Bayes ROC Curve

alt text

Screenshot of output

alt text