Skip to content
Branch: master
Find file History
Pull request Compare This branch is 58 commits ahead, 113 commits behind VowpalWabbit:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Contextual Memory Tree (CMT)

This demo exercises CMT for applications of logarithmic time multiclass classification (online and offline), and logarithmic time multilabel classification.

The datasets for multiclass classification used are ALOI and WikiPara. ALOI has 1000 classes, and each class has in average 100 training examples. WikiPara contains 10000 classes. We consider two versions of WikiPara here: 1-shot version which contains 1 training example per class, and 2-shot version which contains 2 training examples per class.

The datasets for multilabel classification used are RCV1-2K, AmazonCat-13K, and Wiki10-31K from the XML repository.

We refer users to our ICML 2019 paper for detailed datastrutures and algorithms in CMT


python 3

Training Online Contextual Memory Tree on ALOI and WikiPara:


Training Offline Contextual Memory Tree on ALOI, WikiPara, RCV1-2K, AmazonCat-13K and Wiki10-31K:

You can’t perform that action at this time.