rime.models.rnn.RNN

class rime.models.rnn.RNN(item_df, max_item_size=30000, num_hidden=128, nlayers=2, max_epochs=20, gpus=0, truncated_input_steps=256, truncated_bptt_steps=32, batch_size=64, load_from_checkpoint=None, auto_pad_item=True)

Bases: object

__init__(item_df, max_item_size=30000, num_hidden=128, nlayers=2, max_epochs=20, gpus=0, truncated_input_steps=256, truncated_bptt_steps=32, batch_size=64, load_from_checkpoint=None, auto_pad_item=True)

Methods

__init__(item_df[, max_item_size, ...])
fit(D)
transform(D)