DSCKGEC / learn-machine-learn

A one-stop repository for new-comers in Machine Learning and A.I.
MIT License
1 stars 4 forks source link

Improved Naive Bayes Model #37

Closed srini047 closed 2 years ago

srini047 commented 2 years ago

Linked Issue

Closes: #7

Describe the changes you've made

I have improved the existing Naive Bayes model by 1% after analysing the trend in the dataset.

Type of change

What sort of change have you made:

How Has This Been Tested?

  1. Initially plotted the SNS Heat and Cluster Map.
  2. Found that the Uniformity of cell size column is highly correlated which is causing the model to perform poor.
  3. After removing it can see a 1% betterment in the Naive Bayes Model and also significant improvement in other models excluding the logistic regression model.

Checklist:

Screenshots

heat-map

cluster-map

srini047 commented 2 years ago

Closed to fix Merge Conflict.