Fruit Classification

Group members: Hannah Li, Trisha Paul, Puja Subramaniam, Shelby Watson

Blog post detailing our modeling process: https://medium.com/@hannah.li/fruit-classification-1f64d503d3e7

Fruit Classification.ipynb

• contains code to read in images
• perform and visualize PCA and TSNE
• train and test KNN, Decision Tree, and Random Forest

CNN.ipynb

• code to run CNN
• model accuracy
• visualize CNN results

VGG16.ipynb

• code to run VGG16
• model accuracy