Model#

class sdm.models.Model#

Bases: Module, ABC

Base model for foundation models on structured data.

Model extends torch.nn.Module by unified pre-processing and post-processing routines, caching, and ensembling.

supported_feature_stypes: ClassVar[frozenset[Stype]]#

Semantic types supported for input columns in this model.

abstractmethod classmethod default_recipe() → Recipe#

Return the default processing recipe for this model.

Return type:

Recipe

clear() → None#

Clear any cached state.

Return type:

None