https://github.com/maicius/sgm_pk
学习：https://github.com/lancopku/SGM
https://github.com/maicius/sgm_pk
Last synced: about 1 month ago
JSON representation
学习：https://github.com/lancopku/SGM
- Host: GitHub
- URL: https://github.com/maicius/sgm_pk
- Owner: Maicius
- Created: 2019-10-14T09:29:52.000Z (almost 7 years ago)
- Default Branch: master
- Last Pushed: 2019-10-14T09:32:28.000Z (almost 7 years ago)
- Last Synced: 2025-02-27T02:50:48.976Z (over 1 year ago)
- Language: Python
- Homepage:
- Size: 20.5 KB
- Stars: 0
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Sequence Generation Model for Multi-label Classification
- This is the code for our paper *SGM: Sequence Generation Model for Multi-label Classification* [[pdf]](https://arxiv.org/abs/1806.04822)
- Be careful: the provided code is based on the RCV1-V2 dataset. If you need to run the code on other datasets, please correspondly modify all program statements that relate to the specific name of the dataset.
***********************************************************
## Datasets
* RCV1-V2
* AAPD
Two datasets are available at https://drive.google.com/file/d/18-JOCIj9v5bZCrn9CIsk23W4wyhroCp_/view?usp=sharing
***************************************************************
## Requirements
* Ubuntu 16.0.4
* Python 3.5
* Pytorch 0.3.1
***************************************************************
## Reproducibility
We provide the pretrained checkpoints of the SGM model and the SGM+GE model on the RCV1-V2 dataset to help you to reproduce our reported experimental results. The detailed reproduction steps are as follows:
- Please download the RCV1-V2 dataset and checkpoints first by clicking on the link provided above, then put them in the folder *./data/data/*
- Preprocessing: ```python3 preprocess.py ```
- Predict: ```python3 predict.py -gpus id -log log_name```
***************************************************************
## Preprocessing
```
python3 preprocess.py
```
Remember to download the dataset and put them in the folder *./data/data/*
***************************************************************
## Training
```
python3 train.py -gpus id -log log_name
```
****************************************************************
## Evaluation
```
python3 predict.py -gpus id -restore checkpoint -log log_name
```
*******************************************************************
## Citation
If you use the above code or the AAPD dataset for your research, please cite the paper:
```
@inproceedings{YangCOLING2018,
author = {Pengcheng Yang and
Xu Sun and
Wei Li and
Shuming Ma and
Wei Wu and
Houfeng Wang},
title = {{SGM:} Sequence Generation Model for Multi-label Classification},
booktitle = {Proceedings of the 27th International Conference on Computational
Linguistics, {COLING} 2018, Santa Fe, New Mexico, USA, August 20-26,
2018},
pages = {3915--3926},
year = {2018}
}
```