DECAF

DECAF: Deep Extreme Classification with Label Features

@InProceedings{Mittal21,
author = "Mittal, A. and Dahiya, K. and Agrawal, S. and Saini, D. and Agarwal, S. and Kar, P. and Varma, M.",
title = "DECAF: Deep Extreme Classification with Label Features",
booktitle = "Proceedings of the ACM International Conference on Web Search and Data Mining",
month = "March",
year = "2021",
}

SETUP WORKSPACE

mkdir -p ${HOME}/scratch/XC/data mkdir -p ${HOME}/scratch/XC/programs

SETUP DECAF

cd${HOME}/scratch/XC/programs
git clone https://github.com/Extreme-classification/DECAF.git
conda create -f DECAF/decaf_env.yml
conda activate decaf
git clone https://github.com/kunaldahiya/pyxclib.git
cd pyxclib
python setup.py install
cd ../DECAF

DOWNLOAD DATASET

cd${HOME}/scratch/XC/data
gdown --id <dataset id>
unzip *.zip

dataset	dataset id
LF-AmazonTitles-131K	1VlfcdJKJA99223fLEawRmrXhXpwjwJKn
LF-WikiSeeAlsoTitles-131K	1edWtizAFBbUzxo9Z2wipGSEA9bfy5mdX
LF-AmazonTitles-1.3M	1Davc6BIfoTIAS3mP1mUY5EGcGr2zN2pO

RUNNING DECAF

cd${HOME}/scratch/XC/programs/DECAF
chmod +x run_DECAF.sh
./run_DECAF.sh <gpu_id><DECAF TYPE><dataset><folder name>
e.g.
./run_DECAF.sh 0 DECAF LF-AmazonTitles-131K DECAF_RUN
./run_DECAF.sh 0 DECAF-lite LF-AmazonTitles-131K DECAF_RUN

YOU MAY ALSO LIKE

• DeepXML: A Deep Extreme Multi-Label Learning Framework Applied to Short Text Documents
• ECLARE: Extreme Classification with Label Graph Correlations
• GalaXC: Graph Neural Networks with Labelwise Attention for Extreme Classification