keras_pooled_gru: keras pooled gru

Description Usage Arguments Details Value

Description

Word embedding + spatial dropout + (pooled) gated recurrent unit

Usage

1
2
3
keras_pooled_gru(input_dim, embed_dim = 128, seq_len = 50,
  gru_dim = 64, gru_drop = 0.2, bidirectional = F,
  output_fun = "softmax", output_dim = 2)

Arguments

input_dim

Number of unique vocabluary/tokens

embed_dim

Number of word vectors

seq_len

Length of the input sequences

gru_dim

Number of recurrent neurons (default 64)

gru_drop

Recurrent dropout ratio

bidirectional

default is F

output_fun

Output activation function

output_dim

Number of neurons of the output layer

Details

Taken from https://www.kaggle.com/yekenot/pooled-gru-fasttext

Value

keras model


systats/textlearnR documentation built on May 6, 2019, 8:31 p.m.