A deep learning based multi-view multi-label learning baseline implemented with PyTorch.
We use mirflickr as our running example since it is commonly used and has a moderate size. Dataset location: data/mirflickr
- Python 3.7+
- PyTorch 1.7.0
- numpy 1.17.3
- sklearn 0.22.1
Older versions might work as well.
To train the model:
bash script/run_train_mirflickr.sh
To test the model (this .sh will be produced automatically):
bash script/run_test_mirflickr.sh
The seed is 1 by default, but can be changed in the bash file.