Skip to content

billyporter/WeSTClass-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

53 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

WeSTClass-PyTorch

PyTorch Implementation of WeSTClass Weakly-Supervised Neural Text Classification http://chaozhang.org/papers/2018-cikm-westclass.pdf

Usage

usage: main.py [-h] [--data {generate,load}] [--model {rnn,bert,cnn}]
      [--sup_source {labels,keywords,docs}] [--pretrain]
      [--selftrain] [--evaluate] [--with_statistics] [--load_model]
      [--save_docs]

optional arguments:
  -h, --help show this help message and exit
  --data {generate,load}
  --model {rnn,bert,cnn}
  --sup_source {labels,keywords,docs}
  --pretrain
  --selftrain
  --evaluate
  --with_statistics
  --load_model
  --save_docs

Requirements

  • Numpy Version: 1.21.5
  • PyTorch Version: 1.10.0+cu111
  • Python Version: 3.7.12
  • Scikit-Learn Version: 0.22
  • TQDM Version: 4.63.0
  • gensim Version: 4.2.0

About

PyTorch Implementation of WeSTClass

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages