def fit(self, X, y=None):
self._sklearn_model = SKLModel(**self._hyperparams)
if (y is not None):
self._sklearn_model.fit(X, y)
else:
self._sklearn_model.fit(X)
return self
def __init__(self, neg_label=0, pos_label=1, sparse_output=False):
self._hyperparams = {
'neg_label': neg_label,
'pos_label': pos_label,
'sparse_output': sparse_output
}
self._wrapped_model = SKLModel(**self._hyperparams)
def __init__(self):
self._hyperparams = {}
self._wrapped_model = SKLModel(**self._hyperparams)
def __init__(self, classes=None, sparse_output=False):
self._hyperparams = {
'classes': classes,
'sparse_output': sparse_output}
self._wrapped_model = SKLModel(**self._hyperparams)