keras_bi_lstm: keras_bi_lstm

Description Usage Arguments Value

View source: R/keras_models.R

Description

Word embedding + (bidirectional) long short-term memory

Usage

1
2
3
keras_bi_lstm(input_dim, embed_dim = 128, seq_len = 50,
  lstm_dim = 64, lstm_drop = 0.2, dropout = 0.2, output_dim = 1,
  output_fun = "softmax")

Arguments

input_dim

Number of unique vocabluary/tokens

embed_dim

Number of word vectors

seq_len

Length of the input sequences

lstm_dim

Number of recurrent neurons (default 64)

lstm_drop

Recurrent dropout ratio

output_dim

Number of neurons of the output layer

output_fun

Output activation function

Value

keras model


systats/deeplyr documentation built on Oct. 4, 2020, 7:59 p.m.