RunzheYang / TextClassification

CS420 Coursework: Text Classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hierarchical Neural Model for Recommending Articles

CS420 Coursework: Text Classification

In this coursework, I implemented various advanced neural network architectures for a binary text classification task. Based on several successful convolutional neural networks (CNN), I proposed a novel hierarchical neural model, which is compatible with inherent hierarchical properties of documents. The experiment shows that the proposed model outperforms conventional character-level CNNs for text classification on AUC criterion. Even without sufficient training, the stacking tricks integrating existing models help further improve the final performance in the in-class kaggle competition.

Hierarchical Neural Model

In this coursework, I design a hierarchical neural model, which is shown below. For detailed information and evaluation, see my final report here.

Environment

I ran experiments on my PC.

  • Operating System: CentOS Linux release 7.3.1611 (Core)
  • CPU: Intel(R) Core(TM) i5-4460 CPU @ 3.20GHz
  • No GPU
  • Total Memory: 16142596 kB. The code is in Python 2.7 and I implemented neural networks with PyTorch.

About

CS420 Coursework: Text Classification


Languages

Language:Python 76.5%Language:TeX 23.5%