/Deep-MVML

A deep learning based multi-view multi-label learning baseline implemented with PyTorch.

Primary LanguagePython

Deep-MVML

A deep learning based multi-view multi-label learning baseline implemented with PyTorch.

Sample Dataset

We use mirflickr as our running example since it is commonly used and has a moderate size. Dataset location: data/mirflickr

Dependencies

  • Python 3.7+
  • PyTorch 1.7.0
  • numpy 1.17.3
  • sklearn 0.22.1

Older versions might work as well.

Run

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.