simpleml.models.classifiers.external_models

Module Contents

Classes

ClassificationExternalModelMixin

Wrapper class for a pickleable model with expected methods

Attributes

LOGGER
__author__

simpleml.models.classifiers.external_models.LOGGER

simpleml.models.classifiers.external_models.__author__ = Elisha Yadgaran

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(self, *args, **kwargs)

By default fall back to predict method