hparik11 / Cuisine_Classifier

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cuisine Classifier

Installation:

  • conda create -n yummlyenv python=3.6 anaconda
  • source activate yummlyenv
  • conda install pip
  • pip install -r requirement.txt

How to run:

  • First run Yummly_Data_Training.ipynb and generate model files. (Few Model files are too big...)
python3 yummly_cuisine_classification.py cuisine.unlabeled.json.gz
  1. It will ask for your choice of method:
    • 1 for logistic regression,
    • 2 for random forests,
    • 3 for Text CNN.
  2. Based on the choice, it will load the selected model.
  3. After it will perform data loading and prediction of cuisines.
  4. At the end, it will generate a CSV on your local machine.

About

License:Apache License 2.0


Languages

Language:HTML 70.1%Language:Jupyter Notebook 28.8%Language:Python 1.1%