RNN

class daart.models.rnn.RNN(hparams, type='encoder', in_size=None, hid_size=None, out_size=None)[source]

Bases: BaseModel

Methods Summary

forward(x, **kwargs)

Process input data.

Methods Documentation

forward(x, **kwargs)[source]

Process input data.

Parameters:

x (torch.Tensor object) – input data of shape (n_sequences, sequence_length, n_markers)

Returns:

shape (n_sequences, sequence_length, n) where n is the embedding dimension if an encoder, or n_markers if a decoder/predictor

Return type:

torch.Tensor