/Cuisine_Prediction

BUAA SCSE Autumn 2021 Machine Learning Personal Homework

Primary LanguagePythonMIT LicenseMIT

Cuisine_Prediction

GitHub

The technical report can be found here.

Requirement

  • Python >= 3.7
  • torch >= 1.9.0
  • numpy >= 1.17.2
  • matplotlib >= 3.1.1 [optional]
  • scikit-learn >= 0.21.3 [optional]

Preparation

Clone

git clone https://github.com/hiyouga/Cuisine_Prediction.git

Create an anaconda environment:

conda create -n cuisine python=3.7
conda activate cuisine
pip install -r requirements.txt

Usage

Training

python main.py

Ensemble

python main.py --ensemble [1.pt,2.pt,3.pt]

Visualization with t-SNE

python main.py --checkpoint [*.pt]

The checkpoint files can be found in the state_dict folder.

Show help message

python main.py -h

Acknowledgements

This is a personal homework for "Machine Learning" in BUAA Graduate School.

Contact

hiyouga [AT] buaa [DOT] edu [DOT] cn

License

MIT