Skip to content

efikarra/hierarchical-rnn-tf

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

74 Commits
 
 
 
 
 
 
 
 

Repository files navigation

This repository implements several hierarchical and non-hierarchical Deep Learning models in Tensorflow for text classification of documents. It also supports pre-trained word or character embeddings as input to the models.

The module "model/model.py" contains a Feed Forward Neural Network (FFN), a Recurrent Neural Network (RNN) and a Convolutional Neural Network (CNN). The model "model/hierarchical_model.py" contains the following hierarchical models:

  1. A RNN in the word-level with a RNN in the sentence-level.
  2. A CNN in the word-level with a RNN in the sentence-level.
  3. A FFN in the word-level with a RNN in the sentence-level.
  4. A RNN in the word-level, a RNN in the sentence-level and a Conditional Random Field (CRF) as a final layer.

Required packages

  1. Tensorflow 1.8
  2. numpy
  3. scipy
  4. matplotlib

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages