项目作者: minhntm

项目描述 :
Sequence Labeling implemented by RNN/GRU/LSTM using Tensorflow
高级语言: Python
项目地址: git://github.com/minhntm/sequence-labeling.git
创建时间: 2019-10-19T02:49:18Z
项目社区:https://github.com/minhntm/sequence-labeling

开源协议:

下载


Sequence-labeling

Sequence Labeling implemented by RNN/GRU/LSTM using Tensorflow.

Install requirements

  • tensorflow 1.6
  • numpy
  • python 3.6

Model hyperparameters

  • init_scale: the initial value of parameters
  • learning_rate: the initial value of learning rate
  • max_grad_norm: the maximum norm value of gradient
  • num_layers: the layer number of model
  • hidden_size: the node number of hidden layer
  • epoch: the the total number of epochs for training
  • keep_prob: the dropout probability
  • lr_decay: the decay of the learning rate
  • batch_size: the number of inputs
  • model: model type (rnn/gru/lstm)
  • save_path: the folder to save parameter after training
  • data_file: data of model
  • kfold: the k-fold value of cross-validation test method

Evaluation

  • kfold cross-validation method (warning: because of extremely long running time, in deep learning area, we don’t use kfold cross-validation to evaluate model. This is just a funny experiment.)
  • f1 score metric

Training and evaluate model

  1. python rnn_model.py --config_file=config.cfg