fishy.models.deep.lstm

Bidirectional Gated Recurrent Unit (GRU) model for spectral classification.

Classes

class fishy.models.deep.lstm.LSTM(input_dim: int, output_dim: int, hidden_dim: int = 128, num_layers: int = 2, dropout: float = 0.2, **kwargs)

Bases: Module

Bidirectional GRU-based model for spectral data classification.

__init__(input_dim: int, output_dim: int, hidden_dim: int = 128, num_layers: int = 2, dropout: float = 0.2, **kwargs) → None

Initializes the GRU model.

forward(x: Tensor) → Tensor

Forward pass of the GRU model.

s