MomentHyperparams

moment_model.MomentHyperparams · inherits PydanticBaseModel

(No documentation provided)

__init__(self, ...)

(Inherited from PydanticBaseModel)

MomentDataset

moment_model.MomentDataset · inherits Dataset

Dataset class for MOMENT model training and inference.

__init__(self, data: np.ndarray, context_length: int, prediction_length: int)

(No prose summary provided)

Parameter	Type	Default	Description
data	np.ndarray	-	(undocumented)
context_length	int	-	(undocumented)
prediction_length	int	-	(undocumented)

MomentModel

moment_model.MomentModel · inherits BaseModel

MOMENT model wrapper for time series forecasting, extending FoundationModel.

__init__(self, params: Dict[str, Any], settings: Dict[str, Any])

(No prose summary provided)

Parameter	Type	Default	Description
params	Dict[str, Any]	-	(undocumented)
settings	Dict[str, Any]	-	(undocumented)

train(self, y_context: np.ndarray, y_target: np.ndarray, timestamps_context: np.ndarray, timestamps_target: np.ndarray, x_context: Optional[np.ndarray] = None, x_target: Optional[np.ndarray] = None, **kwargs: dict) -> MomentModel

(No prose summary provided)

Parameter	Type	Default	Description
y_context	np.ndarray	-	(undocumented)
y_target	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	dict	-	(undocumented)

Returns: MomentModel (the fitted forecaster; sets self.model_).

predict(self, y_context: np.ndarray, timestamps_context: np.ndarray, timestamps_target: np.ndarray, x_context: Optional[np.ndarray] = None, x_target: Optional[np.ndarray] = None, **kwargs: dict) -> np.ndarray

Make predictions using the trained MOMENT model.

Parameter	Type	Default	Description
y_context	np.ndarray	-	Recent/past target values
timestamps_context	np.ndarray	-	Timestamps for context data (not used)
timestamps_target	np.ndarray	-	Timestamps for target data (used for determining forecast_horizon)
x_context	Optional[np.ndarray]	None	(undocumented)
x_target	Optional[np.ndarray]	None	(undocumented)
**kwargs	dict	-	Additional arguments (unused)

Returns: np.ndarray (Model predictions with shape (channels, forecast_horizon)).