simpleml.models.classifiers.external_models module

class simpleml.models.classifiers.external_models.ClassificationExternalModelMixin

Bases: simpleml.models.external_models.ExternalModelMixin

Wrapper class for a pickleable model with expected methods

Expected to be used as Mixin Class with default methods and ovewritten by the model class if methods exist

Extended from base class to add classification methods

ex:

from some_model_library import ActualModelClass

class WrappedActualModelClass(ActualModelClass, ClassificationExternalModelMixin):

pass

class some_model_libraryActualModelClass(Model, [optional mixins]):

def _create_external_model(self, **kwargs):

return WrappedActualModelClass(**kwargs)

predict_proba(*args, **kwargs)

By default fall back to predict method