Giter Club home page Giter Club logo

cornet's Introduction

CorNet

Correlation Networks for Extreme Multi-label Text Classification

Prerequisites

  • python==3.6.3
  • pytorch==1.2.0
  • torchgpipe==0.0.5
  • click==7.0
  • ruamel.yaml==0.16.5
  • numpy==1.16.2
  • scipy==1.2.1
  • scikit-learn==0.20.3
  • gensim==3.7.2
  • nltk==3.2.4
  • tqdm==4.31.1
  • joblib==0.13.2
  • logzero==1.5.0

Datasets

Pretrained Word Embeddings in gensim format

Run

Preprocess (the EUR-Lex dataset is already tokenized in advance)

./scripts/preprocess_eurlex.sh

or (the other datasets need to be tokenized using NLTK)

./scripts/preprocess_others.sh

Train and evaluate

./scripts/run_models.sh

Baselines

The codes for the baseline models are adapted from the following repositories: XML-CNN, BERT, MeSHProbeNet, and AttentionXML.

cornet's People

Contributors

xunguangxu avatar

Stargazers

junbin avatar avatar shihong avatar Roy Huang avatar Omid Ghahroodi avatar Fu Lin avatar avatar 一只小宏宏 avatar avatar Lorenzo Gianassi avatar avatar Michael Corrado avatar Coding_Writer avatar 乐屿心扉 avatar Wu Jiawei avatar avatar avatar avatar avatar avatar Tobias Senst avatar avatar avatar Zhya avatar Bobobobo avatar avatar shelton-xiaoshuaichen avatar NLPerUndefined avatar avatar wangdui avatar 爱可可-爱生活 avatar felix-wang avatar Zheng Yuan avatar GaoYongWei avatar DajunZhou avatar Yu Zhang avatar haoc_leo avatar yechenchen avatar trayfour avatar avatar Chichi avatar wangxun avatar 李振梁 avatar avatar zzzzz avatar Snehal Patel avatar Bryan (Ning) Xia avatar avatar Jianhui Sun avatar

Watchers

avatar

Forkers

jhnlp shirl3yll sylin117 enockipp dumpmemory codegodliu gk12happy shlshlshk1996 769261443 coolabhishek gaybro8777 goyalkaraniit nibuhao

cornet's Issues

关于CorNet在其他模型上实现无效的疑惑

您好！在阅读您的文章以及相关代码后，我尝试把CorNet同样用于极多标签文本分类的一些模型上，例如近一两年的LightXML，但是我依你所指，在output后添加CorNet，但是发现运行出来的结果不升反降，这当中究竟是不是我在理解文章以及源码的过程中有什么忽略的地方所在呢？烦请您百忙中抽空解答谢谢！

License for the codebase

Hello!
We are planning to re-use the code as baselines for our research. Is it possible for you to add a license (i.e. MIT or BSD) for the codebase?
Thanks!

bertxml

Hi!
well job！
When we use BertXML or BertXMLCornet, do we still need to use Glove?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.