Welcome to hydromodel

A lightweight Python package for hydrological model calibration and evaluation, featuring the XinAnJiang (XAJ) model.

hydromodel is a Python implementation of conceptual hydrological models, with a focus on the XinAnJiang (XAJ) model - one of the most widely-used rainfall-runoff models, especially in China and Asian regions. The package provides comprehensive tools for model calibration, evaluation, and simulation with a unified API design.

Key Features

🏞️ Hydrological Models

• XAJ Model Variants: Standard XAJ and optimized versions (xaj_mz with Muskingum routing, xaj_slw)
• GR Models: GR1A, GR2M, GR3J, GR4J, GR5J, GR6J
• Other Models: HYMOD, DHF(Dahuofang model)
• Extensible Framework: Easy to add custom models

🔧 Calibration Algorithms

• SCE-UA: Shuffled Complex Evolution (robust, recommended for global optimization)
• GA: Genetic Algorithm with DEAP (flexible, handles complex parameter landscapes)
• scipy: L-BFGS-B, SLSQP, and other gradient-based methods (fast for smooth objectives)

📊 Evaluation & Analysis

• Comprehensive Metrics: NSE, KGE, RMSE, PBIAS, FHV, FLV, FMS
• Multi-Basin Support: Efficient calibration and evaluation for multiple basins simultaneously
• Time Series Analysis: Flood event extraction and characterization
• Visualization: Automated plotting of simulation results and metrics

🗄️ Data Integration

• CAMELS Datasets: Seamless support for 11 CAMELS variants via hydrodataset
• Custom Data: Flexible support for user data via hydrodatasource
• Flood Event Data: Specialized support for discrete flood event datasets
• Standardized Format: Unified data interface across all data sources

🚀 Developer-Friendly

• Unified API: Consistent interfaces for calibration, evaluation, and simulation
• Configuration-Based: YAML configuration for reproducibility
• Progress Tracking: Real-time progress display and intermediate results saving
• Standardized Results: All algorithms save results in unified JSON + CSV format

Quick Start

from hydromodel.trainers.unified_calibrate import calibrate
from hydromodel.trainers.unified_evaluate import evaluate

# Configuration for calibration
config = {
    "data_cfgs": {
        "data_source_type": "camels_us",
        "basin_ids": ["01013500"],
        "train_period": ["1985-10-01", "1995-09-30"],
        "test_period": ["2005-10-01", "2014-09-30"],
        "warmup_length": 365,
        "variables": ["precipitation", "potential_evapotranspiration", "streamflow"]
    },
    "model_cfgs": {
        "model_name": "xaj_mz",
    },
    "training_cfgs": {
        "algorithm_name": "SCE_UA",
        "algorithm_params": {"rep": 5000, "ngs": 1000},
        "loss_config": {"type": "time_series", "obj_func": "RMSE"},
        "output_dir": "results",
        "experiment_name": "my_experiment",
    },
    "evaluation_cfgs": {
        "metrics": ["NSE", "KGE", "RMSE"],
    },
}

# Run calibration
results = calibrate(config)

# Evaluate on test period
evaluate(config, param_dir="results/my_experiment", eval_period="test")

Or use command-line scripts:

# 1. Calibration
python scripts/run_xaj_calibration.py --config configs/example_config.yaml

# 2. Evaluation
python scripts/run_xaj_evaluate.py --calibration-dir results/xaj_mz_SCE_UA --eval-period test

# 3. Simulation (no calibration required!)
python scripts/run_xaj_simulate.py \
    --config configs/example_simulate_config.yaml \
    --param-file configs/example_xaj_params.yaml \
    --plot

# 4. Visualization
python scripts/visualize.py --eval-dir results/xaj_mz_SCE_UA/evaluation_test

Installation

Quick Installation

pip install hydromodel hydrodataset

Or using uv (faster):

uv pip install hydromodel hydrodataset

From Source

git clone https://github.com/OuyangWenyu/hydromodel.git
cd hydromodel
uv sync --all-extras

For detailed installation instructions, see the Installation Guide.

Why hydromodel?

For Researchers: - Battle-tested XAJ implementations used in published research - Configuration-based workflow ensures reproducibility - Easy to extend with new models or calibration algorithms - Lightweight and fast - perfect for parameter sensitivity studies

For Practitioners: - Simple YAML configuration, minimal coding required - Handles multi-basin calibration efficiently - Integration with global CAMELS datasets (11 variants) - Clear documentation and examples

Compared to other packages: - vs. SWAT/VIC: Lighter weight, Python-native, faster iteration - vs. pySTREPS: Focus on conceptual rainfall-runoff models - vs. custom scripts: Well-tested with unified interfaces

Documentation Structure

• Installation Guide - Detailed installation instructions for all platforms
• Quick Start - Get started in 5 minutes
• Usage Guide - Comprehensive tutorials and examples
• Data Guide - How to prepare and use different data sources
• API Reference - Complete API documentation
• Model Documentation - Detailed model descriptions (XAJ, DHF)
• Contributing - How to contribute to the project
• FAQ - Frequently asked questions
• Changelog - Version history and updates

Use Cases

1. Model Calibration

Calibrate hydrological models on CAMELS datasets or custom data with various algorithms:

# Use SCE-UA for global optimization
config["training_cfgs"]["algorithm_name"] = "SCE_UA"
results = calibrate(config)

# Or use GA for flexible optimization
config["training_cfgs"]["algorithm_name"] = "GA"
results = calibrate(config)

2. Multi-Basin Evaluation

Efficiently calibrate and evaluate multiple basins:

config["data_cfgs"]["basin_ids"] = ["01013500", "01022500", "01030500"]
results = calibrate(config)
evaluate(config, param_dir="results/my_experiment", eval_period="test")

3. Parameter Sensitivity Analysis

Run simulations with custom parameter sets:

from hydromodel.trainers.unified_simulate import UnifiedSimulator

# Test different parameter values
for k_value in [0.5, 0.75, 1.0]:
    parameters = {..., "K": k_value, ...}
    simulator = UnifiedSimulator(model_config, basin_config)
    results = simulator.simulate(inputs, qobs, warmup_length=365)
    # Analyze results

4. Flood Event Analysis

Extract and calibrate on flood events:

config = {
    "data_cfgs": {
        "data_source_type": "floodevent",
        "time_unit": ["3h"],
        "variables": ["prcp", "PET", "streamflow"],
        ...
    },
    ...
}
results = calibrate(config)

Supported Models

Model	Description	Parameters	Routing
xaj	Standard XinAnJiang model	15	Linear reservoir
xaj_mz	XAJ with Muskingum routing	15	Muskingum
xaj_slw	XAJ with SLW routing	26	Storage-lag-weighted
gr4j	GR4J rainfall-runoff model	4	Unit hydrograph
hymod	HYMOD model	5	Nash cascade
dhf	Dahuofang model	18	Custom

For detailed model documentation, see XAJ Model and DHF Model.

Calibration Algorithms

Algorithm	Type	Strengths	Best For
SCE-UA	Global	Robust, reliable convergence	General purpose, recommended
GA	Global	Flexible, handles discontinuities	Complex parameter landscapes
scipy	Local	Fast, gradient-based	Smooth objectives, refinement

Data Sources

CAMELS Datasets

Support for 11 CAMELS variants worldwide:

• CAMELS-US (671 basins)
• CAMELS-AUS (222 basins)
• CAMELS-BR (897 basins)
• CAMELS-CL (516 basins)
• CAMELS-GB (671 basins)
• And more...

Custom Data

Use your own data with selfmadehydrodataset format:

my_basin_data/
├── attributes/
│   └── attributes.csv
├── shapes/
│   └── basins.shp
├── timeseries/
│   ├── 1D/
│   │   ├── basin_001.csv
│   │   └── basin_002.csv
│   └── 1D_units_info.json

See Data Guide for complete specifications.

Performance

• Fast calibration: Optimized algorithms with numba JIT compilation
• Memory efficient: Handles large datasets with chunked processing
• Parallel support: Multi-basin calibration runs independently
• Progress tracking: Real-time monitoring of long-running calibrations

References

XAJ Model: - Zhao, R.J., 1992. The Xinanjiang model applied in China. Journal of Hydrology, 135(1-4), pp.371-381.

Calibration Algorithms: - Duan, Q., et al., 1992. Effective and efficient global optimization for conceptual rainfall-runoff models. Water Resources Research, 28(4), pp.1015-1031. (SCE-UA)

Related Projects: - hydrodataset - CAMELS and other datasets - hydrodatasource - Data preparation utilities - torchhydro - PyTorch-based hydrological models

Citation

If you use hydromodel in your research, please cite:

@software{hydromodel,
  author = {Ouyang, Wenyu},
  title = {hydromodel: A Python Package for Hydrological Model Calibration},
  year = {2025},
  url = {https://github.com/OuyangWenyu/hydromodel}
}

License & Credits

• License: GNU General Public License v3.0
• Author: Wenyu Ouyang
• Documentation: https://OuyangWenyu.github.io/hydromodel
• Source Code: https://github.com/OuyangWenyu/hydromodel

Getting Help

• Documentation: Browse the complete documentation
• Issues: Report bugs or request features at GitHub Issues
• Discussions: Ask questions at GitHub Discussions
• Email: wenyuouyang@outlook.com

Contributing

Contributions are welcome! See the Contributing Guide for details on:

• Reporting bugs
• Suggesting features
• Submitting pull requests
• Code style and testing guidelines

Community

Join our growing community:

• ⭐ Star the project on GitHub
• 🐛 Report issues and bugs
• 💡 Suggest new features
• 📖 Improve documentation
• 🔧 Contribute code

---

Ready to get started? Head to the Quick Start Guide or Installation Guide!