Repo for the CS6190 PML Course Project
- Python 3.6
- Pytorch
- 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
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}
}