esem.model_adaptor.SKLearnModel

class esem.model_adaptor.SKLearnModel(model)

A wrapper around scikit-learn models.

__init__(model)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(model)	Initialize self.
predict(*args, **kwargs)	This is either the tf model which can be called directly, or a generator over the model.predict (in tf, so it’s quick).
train(training_params, training_data[, verbose])	Train the RF model.