XML-CNN

KO | EN

gitlite — search

#javascript #react #android #rust #linux #docker #java #awesome-list #vue #security #pytorch #mcp #openai #open-source

(★ 64)

Pytorch implementation of the paper Deep learning for extreme multi-label text classification

XML-CNN 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

This repo is from my past life. I am now co-founder and CTO @ [Merlin](getmerlin.in). Check out our [blog](getmerlin.in/blog) and [extension](https://chromewebstore.google.com/detail/merlin-1-click-access-to/camppjleccjaphfdbohjdohecfnoikec)

# XML-CNN
  Pytorch implementation of the paper [Deep Learning for Extreme Multi-label Text Classification](http://nyc.lti.cs.cmu.edu/yiming/Publications/jliu-sigir17.pdf) with dynamic pooling

## Dependencies

* NLTK (stopwords)
* Pytorch >= 0.3.1
* Gensim
* Matplotlib

![](https://raw.githubusercontent.com/siddsax/XML-CNN/master/cnn.jpg)

Directory Structure:

Sample dataset RCV can be downloaded from [here](http://cse.iitk.ac.in/users/siddsax/rcv.p). Trained model on RCV1 can be downloaded from below.

[Trained model](http://cse.iitk.ac.in/users/siddsax/rcvSaved.pt)

Precision Scores

| 1 | 2 | 3 | 4 | 5 | 
|---|---|---|---|---|
|0.96 | 0.8848 | 0.7809 | 0.6436 | 0.5457 |

Note: The scores are slighltly higher as the tested dataset is part of the full test dataset.

Procedure to train and test the model is as follows. The code by default doesn't plot graphs. Though that can be enabled wth a visdom server running [[Visdom](https://github.com/facebookresearch/visdom)] and turing on flag --d
```bash
python main.py # train a model
python main.py --mn=rcv # train a model and save in directory rcv [inside saved_models]
```
This will create multiple files inside the folder ```saved_models/rcv``` in the above case. Checkpoints are saved after every 
```save_step``` epochs, this can be changed with ``--ss`` option in command line. Also a checkpoint is made according to best test precision@1 score and best training batch precision@1.

```bash
python main.py --lm=$DESTINATION OF SAVED MODEL # This resumes trainign from the given checkpoint
```

In order to test the model run
```bash
python main.py --lm=$DESTINATION OF SAVED MODEL --tr=0 
```

This will first print the training error and then the test error while also saving a score_matrix.mat in the folder the model is loaded that can be used to run test scripts from [here](https://drive.google.com/open?id=0B3lPMIHmG6vGN0hSQjFJUHZ0YTg)

XML-CNN

XML-CNN 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

XML-CNN

XML-CNN 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README