EA-LSTM¶

Overview¶

neuralhydrology_ealstm complements the plain LSTM adapter with a lightweight static gating path inspired by EA-LSTM style hydrology models.

At a Glance¶

Hazard Family

Flood

Public catalog grouping used for this model.

Maturity

Implemented

Catalog maturity label used on the index page.

Tasks

Streamflow

Benchmark Family

Flood Benchmark

Primary benchmark-family link used for compatible evaluation coverage.

Description¶

neuralhydrology_ealstm complements the plain LSTM adapter with a lightweight static gating path inspired by EA-LSTM style hydrology models.

It keeps the same graph-temporal input contract as the rest of the flood streamflow roadmap.

Benchmark Compatibility¶

Primary benchmark family: Flood Benchmark

Mapped benchmark ecosystems: WaterBench

External References¶

Paper: Towards learning universal, regional, and local hydrological behaviors via machine learning applied to large-sample datasets | Repo: Repository

Registry Name¶

Primary entrypoint: neuralhydrology_ealstm

Supported Tasks¶

Streamflow

Programmatic Use¶

import torch
from pyhazards.models import build_model

model = build_model(name="neuralhydrology_ealstm", task="regression", input_dim=2, out_dim=1)
preds = model({"x": torch.randn(1, 4, 6, 2)})
print(preds.shape)

Notes¶

This adapter focuses on the entity-aware gating contract, not exact repo parity.