TimeH2O
Collection
Foundation models for time series • 2 items • Updated • 6
TimeH2O-v1 is a 108M-parameter time series foundation model for zero-shot probabilistic forecasting, with support for covariates.
📰 Blog · 💻 GitHub (coming soon) · 📄 Paper (coming soon)
(a) Architecture · (b) Training batch · (c) Transformer block.
TimeH2O-v1 achieves state-of-the-art zero-shot results on GIFT-Eval and fev-bench.
| Benchmark | Metric | TimeH2O-v1 | Timestral-209M | t0-gamma | TimesFM-3 | Chronos-2 |
|---|---|---|---|---|---|---|
| GIFT-Eval | CRPS ↓ | 0.441 | 0.446 | 0.452 | 0.456 | 0.485 |
| MASE ↓ | 0.654 | 0.658 | 0.661 | 0.667 | 0.698 | |
| fev-bench | SQL skill score ↑ | 50.83 | 50.39 | 49.42 | 48.99 | 47.28 |
| MASE skill score ↑ | 39.95 | 39.49 | 37.99 | 37.82 | 35.50 |
| Model | Parameters | Covariates | Best for |
|---|---|---|---|
| TimeH2O-v1 | 108M | ✅ | Max accuracy |
| TimeH2O-v1-tiny | 5.8M | ❌ | Max throughput |
pip install h2ofm
import numpy as np
from h2ofm import TimeH2O
model = TimeH2O.from_pretrained("h2oai/timeh2o-v1")
forecast = model.forecast(np.sin(np.arange(200) / 10), horizon=24)
forecast.point # (24,), median forecast; also forecast.median
forecast.quantiles # (24, 9), default levels 0.1 through 0.9
forecast.quantile(0.9) # (24,)
Non-commercial. See LICENSE.
Datasets overlapping with fev-bench, GIFT-Eval, and TIME were excluded from training.