The technical report can be found here.
- Python >= 3.7
- torch >= 1.9.0
- numpy >= 1.17.2
- matplotlib >= 3.1.1 [optional]
- scikit-learn >= 0.21.3 [optional]
git clone https://github.com/hiyouga/Cuisine_Prediction.gitconda create -n cuisine python=3.7
conda activate cuisine
pip install -r requirements.txtpython main.pypython main.py --ensemble [1.pt,2.pt,3.pt]python main.py --checkpoint [*.pt]The checkpoint files can be found in the state_dict folder.
python main.py -hThis is a personal homework for "Machine Learning" in BUAA Graduate School.
hiyouga [AT] buaa [DOT] edu [DOT] cn
MIT