Skip to content

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
  • y

Returns

self

predict_one

Predict the output of features x.

Parameters

  • x

Returns

The prediction.