Chronos How to use Mattral/chronos-2-small with Chronos:
pip install chronos-forecasting
import pandas as pd
from chronos import BaseChronosPipeline
pipeline = BaseChronosPipeline.from_pretrained("Mattral/chronos-2-small", device_map="cuda")
# Load historical data
context_df = pd.read_csv("https://autogluon.s3.us-west-2.amazonaws.com/datasets/timeseries/misc/AirPassengers.csv")
# Generate predictions
pred_df = pipeline.predict_df(
context_df,
prediction_length=36, # Number of steps to forecast
quantile_levels=[0.1, 0.5, 0.9], # Quantiles for probabilistic forecast
id_column="item_id", # Column identifying different time series
timestamp_column="Month", # Column with datetime information
target="#Passengers", # Column(s) with time series values to predict
) 
