Skip to content

Regression

Use KFCRegressor for continuous target variables.

from kfc_procedure import KFCRegressor

model = KFCRegressor(
    divergences=["euclidean"],
    local_model="linear_regression",
    combiner="mean",
    n_clusters=3,
    random_state=42,
)
model.fit(X_train, y_train)
y_pred = model.predict(X_test)

Regression combiners

Combiner Behavior
mean arithmetic mean across prediction columns
weighted_mean OLS learns weights over prediction columns
stacking_regressor meta-regressor over prediction matrix
gradientcobra COBRA kernel-weighted aggregation
mixcobra COBRA aggregation using input and prediction spaces

Metrics

from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score
print(mean_absolute_error(y_test, y_pred))
print(mean_squared_error(y_test, y_pred))
print(r2_score(y_test, y_pred))