https://github.com/hiyouga/cuisine_prediction
BUAA SCSE Autumn 2021 Machine Learning Personal Homework
https://github.com/hiyouga/cuisine_prediction
attention-mechanism classification convolutional-neural-networks data-augmentation mixup text-classification visualization
Last synced: 3 months ago
JSON representation
BUAA SCSE Autumn 2021 Machine Learning Personal Homework
- Host: GitHub
- URL: https://github.com/hiyouga/cuisine_prediction
- Owner: hiyouga
- License: mit
- Created: 2021-12-19T10:13:34.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2021-12-31T03:56:29.000Z (almost 4 years ago)
- Last Synced: 2025-07-06T12:06:08.616Z (4 months ago)
- Topics: attention-mechanism, classification, convolutional-neural-networks, data-augmentation, mixup, text-classification, visualization
- Language: Python
- Homepage:
- Size: 2.29 MB
- Stars: 3
- Watchers: 2
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Cuisine_Prediction
The technical report can be found [here](technical_report.pdf).
## 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
```bash
git clone https://github.com/hiyouga/Cuisine_Prediction.git
```
### Create an anaconda environment:
```bash
conda create -n cuisine python=3.7
conda activate cuisine
pip install -r requirements.txt
```
## Usage
### Training
```sh
python main.py
```
### Ensemble
```sh
python main.py --ensemble [1.pt,2.pt,3.pt]
```
### Visualization with t-SNE
```sh
python main.py --checkpoint [*.pt]
```
The checkpoint files can be found in the `state_dict` folder.
### Show help message
```sh
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