Skip to content

wphogan/wine_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Wine Classifier

Instructions:

  • Install requirements: pip install -r requirements.txt
  • Download BERT base model (uncased) and place it in /bert
  • To run train.py: python train.py
  • To run train.py in debug mode: python train.py --debug True

Run best model:

  • To run the best saved model, download the checkpoint, unzip it, and place it in /saved_models.
  • Then, run python eval.py.

Misc:

  • Logs for any given run are kept in /log
  • The preprocessing script in /preprocessing combines the raw wine data and splits it randomly into train/dev/test splits (80/10/10).

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages