Repository navigation
v0.0.35
Features
-
PatchTST-FM family checkpoints: Document and compare IBM Research and Granite PatchTST-FM releases via the existing
PatchTSTFMclass (foundationforecast>=0.1.8,granite-tsfm>=0.3.8). Passrepo_idto select the checkpoint:ibm-research/patchtst-fm-r1(default) — CC-BY-NC-SA-4.0 (non-commercial research weights)ibm-granite/granite-timeseries-patchtst-fm-r1— Apache-2.0ibm-granite/granite-timeseries-patchtst-fm-r2— Apache-2.0 / OpenMDW 1.0 (dual-licensed; commercial-friendly)
import pandas as pd from timecopilot.models.foundation.patchtst_fm import PatchTSTFM df = pd.read_csv( "https://timecopilot.s3.amazonaws.com/public/data/air_passengers.csv", parse_dates=["ds"], ) model = PatchTSTFM( repo_id="ibm-granite/granite-timeseries-patchtst-fm-r2", alias="Granite-PatchTST-FM-r2", context_length=2048, ) fcst_df = model.forecast(df, h=12, freq="MS", level=[80, 90])
License note: Only the IBM Research checkpoint (
ibm-research/patchtst-fm-r1) is non-commercial (CC-BY-NC-SA-4.0). Granite PatchTST-FM r1 and r2 use permissive open licenses suitable for commercial use (see model cards above). -
PatchTST-FM family notebook: Added
patchtst-fm-familyexample notebook comparing IBM Researchpatchtst-fm-r1with Granitepatchtst-fm-r1andpatchtst-fm-r2, including prediction intervals and evaluation metrics.
Dependencies
- Bumped
foundationforecastfrom>=0.1.7to>=0.1.8(Granite PatchTST-FM r2 viagranite-tsfm>=0.3.8).
Full Changelog: v0.0.34...v0.0.35