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

    Typebase.Regressor

    Regression model to wrap.

  • func

    Typetyping.Callable

    A function modifying the target before training.

  • inverse_func

    Typetyping.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: 10.999752

Methods

learn_one

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

Parameters

  • x
  • y

predict_one

Predict the output of features x.

Parameters

  • x

Returns

The prediction.