Skip to content

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

Notifications You must be signed in to change notification settings

A11en0/Deep-MVML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages