Skip to content

manasiladdha/ImageClassifierUsingScalaAndDeepLearning

Repository files navigation

CSYE7200_Scala_Project_Team7

Yelp Image Classification Model - using scala, deeplearning4j Team7_UI - using Play and Scala

Build a predictive model which classify user-submitted photos and automatically tag restaurants with multiple labels

  • 0: good_for_lunch
  • 1: good_for_dinner
  • 2: takes_reservations
  • 3: outdoor_seating
  • 4: restaurant_is_expensive

Infrastructure used in this project:

  • User Interface – PlayFramework, Scala Controllers, Lightbend Activator JavaScript, Bootstrap, CSS, HTML
  • Database – In memory database (MySQL) and Slick for database querying
  • Apache Spark – Scala Integration – Convolutional Neural Network Algorithm implemented in Scala using DeepLearning4J.
  • Data Cleaning & Preparation – Scala, MS Excel

Magnitude of Dataset:

  • train_photos.tgz - photos of the training set (234,545 images)
  • test_photos.tgz - photos of the test set (500 images)
  • train_photo_to_biz_ids.csv - maps the photo id to business id (234,545 rows)
  • test_photo_to_biz_ids.csv - maps the photo id to business id ( 500 rows)
  • train.csv - main training dataset. Includes the business id's, and their corresponding labels. (1996 rows) 

About

Yelp Image Classification Model

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •