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
connCAS

Specifies the CAS connection object.

model_tablestring, optional

Specifies the name of CAS table to store the model.

neuronsint, optional

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

n_blocksint, optional

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

max_output_lengthint, optional

Specifies the maximum number of tokens to generate Default: 15

rnn_typestring, optional

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

Returns
Sequential