abhi1kumar / cs6190_pml_project Goto Github PK

Repo for the CS6190 PML Course Project

1. Python 3.6
2. Pytorch
3. Cuda 10.0

virtualenv --python=/usr/bin/python3.6 py36
source py36/bin/activate
pip install torch torchvision

Type the following command in terminal

git clone [email protected]:abhi1kumar/pml.git

We need to make some extra directories to store data and models

mkdir data
mkdir model

Download the following datasets and move it to data folder

1. Bibtex

Please refer the following papers:

@inproceedings{gaure2017probabilistic,
  title={A probabilistic framework for zero-shot multi-label learning},
  author={Gaure, Abhilash and Gupta, Aishwarya and Verma, Vinay Kumar and Rai, Piyush},
  booktitle={UAI},
  year={2017}
}