PatchtstGraniteModel

chronax.models.patchtst_granite_model.PatchtstGraniteModel · inherits BaseForecaster

PatchTST Granite variant. Inherits full logic; hf_model_name from settings.

train(self, *args, **kwargs)

Override to pass config with n_head from num_attention_heads for PatchTSTFM compatibility.

Parameter	Type	Default	Description
*args	tuple	-	(undocumented)
**kwargs	dict	-	(undocumented)

Returns: Self (the fitted forecaster; sets self.is_fitted).