Skip to content

Add pretrained weights#262

Merged
grey-nearing merged 2 commits into
mainfrom
gsnearing-pretrained-weights
Jun 16, 2026
Merged

Add pretrained weights#262
grey-nearing merged 2 commits into
mainfrom
gsnearing-pretrained-weights

Conversation

@grey-nearing

@grey-nearing grey-nearing commented Jun 16, 2026

Copy link
Copy Markdown
Collaborator

Add pre-trained model weights

Introduce two sets of pre-trained model runs (Mean Embedding Forecast
LSTM) based on the Google Flood Hub architecture to support transfer
learning, local fine-tuning, and Prediction in Ungauged Basins (PUB).

Models included:

  • google-floodhub-base (google-floodhub-settings-55-epochs): Full basin baseline capturing global
    hydrological variety.
  • google-floodhub-nse-filtered ('google-floodhub-settings-55-epochs-nse-filtered-0.5-85-epochs'): High-skill baseline filtered
    for catchments with historical NSE > 0.5.

Each model runtime release includes weights (.pt), optimizer states,
pre-computed global data scalers, and the exact configuration files
for full reproducibility.

These models were trained on the FULL historical data period (1982-2023)
with NO temporal holdout/test split. Evaluation on this historical timeline
will yield fundamentally invalid, artificially inflated metrics due to
in-sample data leakage.

Appropriate use cases include:

  • Local fine-tuning / transfer learning (with lower learning rates)
  • Spatial generalization (PUB) on entirely excluded basins
  • Forward-looking inference on post-2023 data periods

@grey-nearing grey-nearing requested a review from amitmarkel as a code owner June 16, 2026 16:55
@google-cla

google-cla Bot commented Jun 16, 2026

Copy link
Copy Markdown

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

@PatrickGoogle PatrickGoogle left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@grey-nearing grey-nearing force-pushed the gsnearing-pretrained-weights branch from 3e7a60a to 215ae1f Compare June 16, 2026 16:59
@grey-nearing grey-nearing force-pushed the gsnearing-pretrained-weights branch from 215ae1f to 804dcf0 Compare June 16, 2026 17:02
@grey-nearing grey-nearing merged commit affbaaa into main Jun 16, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants