opendataval.model.RegressionSkLearnWrapper#
- class opendataval.model.RegressionSkLearnWrapper(base_model, *args, **kwargs)#
Wrapper for sk-learn regression models.
Example:
wrapped = RegressionSkLearnWrapper(LinearRegression)
Parameters#
- base_modelBaseModel
Any sk-learn model that supports
sample_weights
- __init__(base_model, *args, **kwargs)#
Methods
__init__(base_model, *args, **kwargs)clone()Clone Model object.
fit(x_train, y_train, *args[, sample_weight])Fits the model on the training data.
predict(x)Predict values from sk-learn regression model.
Attributes
Models