src.models.model_factory

Functions

get_dl_model(base_model_name, ...)

Assembles and returns a combined PyTorch model.

get_poker_dl_model(model_name, **kwargs)

Returns a PyTorch Lightning model for the Poker Hand dataset.

get_sklearn_model(model_name, is_regression)

Returns a standard scikit-learn model.

Classes

OrdinalSVC([C, kernel, gamma, random_state])

Support Vector Classifier adapted for Ordinal Regression using the cumulative link approach.