TargetTransformRegressor

Modifies the target before training.

The user is expected to check that func and inverse_func are coherent with each other.

Parameters

• regressor (base.Regressor)

  Regression model to wrap.

• func (Callable)

  A function modifying the target before training.

• inverse_func (Callable)

  A function to return to the target's original space.

Examples

>>> import math
>>> from river import compose
>>> from river import datasets
>>> from river import evaluate
>>> from river import linear_model
>>> from river import metrics
>>> from river import preprocessing

>>> dataset = datasets.TrumpApproval()
>>> model = (
...     preprocessing.StandardScaler() |
...     compose.TargetTransformRegressor(
...         regressor=linear_model.LinearRegression(intercept_lr=0.15),
...         func=math.log,
...         inverse_func=math.exp
...     )
... )
>>> metric = metrics.MSE()

>>> evaluate.progressive_val_score(dataset, model, metric)
MSE: 8.759624

Methods

learn_one

Fits to a set of features x and a real-valued target y.

Parameters

• x (dict)
• y (numbers.Number)

Returns

Regressor: self

predict_one

Predict the output of features x.

Parameters

• x (dict)

Returns

Number: The prediction.