Skip to content

Decision tree and logistic regression models that predict fruit names based on mass, width, height, and color.

Notifications You must be signed in to change notification settings

benjaminhdinh/fruit-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Fruit Classification Using Logistic Regression & Decision Trees

The data set is slightly permutated and is found here: https://raw.githubusercontent.com/lkyin/ECS189L/main/fruits.csv

Summary:

  • Built a logistic regression classifier to predict the fruit_name variable in terms of the predictors mass, width, height, and color_score.
  • Built a decision tree classifier to predict the fruit_name variable in terms of the predictors mass, width, height, and color_score.
  • Graphed the decision boundaries for the two models above and determined overall performances.

About

Decision tree and logistic regression models that predict fruit names based on mass, width, height, and color.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published