TinyTimeMixerR1Hyperparams

tiny_time_mixer_r1_model.TinyTimeMixerR1Hyperparams · inherits PydanticBaseModel

A placeholder class for defining TinyTimeMixer R1 hyperparameters.

TinyTimeMixerR1Model

tiny_time_mixer_r1_model.TinyTimeMixerR1Model · inherits BaseModel

A wrapper model utilizing the sktime implementation of TinyTimeMixer (TTM) for time series forecasting, handling context length constraints and covariate alignment.

__init__(self, params, settings)

Initialize TinyTimeMixer model.

Parameter	Type	Default	Description
params	Dict[str, Any]	-	Model parameters dictionary
settings	Dict[str, Any]	-	Settings dictionary containing device, python_version, etc.

train(self, y_context, y_target, timestamps_context, timestamps_target, x_context=None, x_target=None, **kwargs) -> TinyTimeMixerR1Model

Train/fine-tune the foundation model on given data.

Parameters:

Parameter	Type	Default	Description
y_context	np.ndarray	-	Past target values - training data during tuning time, training + validation data during testing time
y_target	np.ndarray	-	Future target values - validation data during tuning time, None during testing time (avoid data leakage)
timestamps_context	np.ndarray	-	Timestamps for y_context (optional)
timestamps_target	np.ndarray	-	Timestamps for y_target (optional)
x_context	Optional[np.ndarray]	None	(undocumented)
x_target	Optional[np.ndarray]	None	(undocumented)
**kwargs		-	Additional keyword arguments

Returns: Self (The fitted model instance).

predict(self, y_context, timestamps_context, timestamps_target, x_context=None, x_target=None, **kwargs)

(No summary provided in docstring)

Parameters:

Parameter	Type	Default	Description
y_context	np.ndarray	-	(undocumented)
timestamps_context	np.ndarray	-	(undocumented)
timestamps_target	np.ndarray	-	(undocumented)
x_context	Optional[np.ndarray]	None	(undocumented)
x_target	Optional[np.ndarray]	None	(undocumented)
**kwargs		-	(undocumented)

Returns: np.ndarray (Forecast array of shape (forecast_horizon, num_targets)).