Google Research 开源用于 AI 洪水预测的水文建模框架

Google Research 已在 GitHub 开源其水文建模框架，用于帮助国家气象和水文服务机构把 AI 洪水预测整合进自身工作流。该 Python 包基于 PyTorch，实现了驱动 Google Flood Hub 的河流预测模型，包含基于 LSTM 的模型架构和可用 Caravan 数据集训练的训练流程，并支持加入本地数据进行训练或微调。