Skip to content

Latest commit

 

History

History
35 lines (27 loc) · 853 Bytes

README.md

File metadata and controls

35 lines (27 loc) · 853 Bytes
Raw

Demo Code for the paper "Contrastive Learning with Complex heterogeneity" accepted by KDD 2022.

Dependencies

  • PyTorch
  • Pandas
  • CV2
  • Sklearn

Command

Multi-view Multi-label setting on Scene dataset

python main.py -d scene -m linear -alpha 0.7 -beta 0.02 -e 200 -hid 100

Multi-view Multi-class setting on XRMB dataset

python main.py -d xrmb -m linear -alpha 5 -beta 0.01 -e 300 -hid 100

Multi-view Multi-class setting on MNIST dataset

python main.py -d mnist -m cnn -alpha 0.1 -beta 1 -e 300 -hid 200

Multi-view Multi-label setting on CelebA dataset

python main.py -d celeba -m vgg -alpha 0.05 -beta 0.1 -e 300

More datasets