MomentLargeHyperparams

moment_large_model.MomentLargeHyperparams · inherits PydanticBaseModel

(undocumented)

__init__

(Inherited from PydanticBaseModel)

MomentLargeDataset

moment_large_model.MomentLargeDataset · inherits Dataset

Dataset class for MOMENT model training and inference.

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

(undocumented)

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

MomentLargeModel

moment_large_model.MomentLargeModel · inherits BaseModel

MOMENT model wrapper for time series forecasting, extending FoundationModel.

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

(undocumented)

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) -> MomentLargeModel

(undocumented)

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: MomentLargeModel (the fitted 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 (forecast_horizon, num_targets)).