dlpy.applications.TextGeneration

dlpy.applications.TextGeneration(conn, model_table='text_generator', neurons=10, max_output_length=15, n_blocks=3, rnn_type='gru')

Generates a text generation model.

Parameters:

conn : CAS

Specifies the CAS connection object.

model_table : string, optional

Specifies the name of CAS table to store the model.

neurons : int, optional

Specifies the number of neurons to be in each layer.
Default: 10

n_blocks : int, optional

Specifies the number of bidirectional blocks to be added to the model.
Default: 3

max_output_length : int, optional

Specifies the maximum number of tokens to generate
Default: 15

rnn_type : string, optional

Specifies the type of the rnn layer.
Default: GRU
Valid Values: RNN, LSTM, GRU

Returns:

Sequential