MachineLearning/gru library

🔁 Gated Recurrent Unit (GRU)

A compact GRU cell and sequence model exposing a tidy API for sequence modeling. Designed to be generic and extendable to production needs (e.g., dropout, multilayer stacking), with detailed docstrings describing inputs, outputs, and error cases.

Contract:

  • Input: sequence batch X (n x t x m)
  • Output: sequence-level predictions or hidden states
  • Errors: throws ArgumentError for invalid shapes

Classes

GRU
Minimal GRU wrapper delegating the final mapping to an ANN readout.