-
Notifications
You must be signed in to change notification settings - Fork 425
Closed
Description
After running the following code:
import numpy as np
import pandas as pd
from supervised.automl import AutoML
df = pd.read_csv("./tests/data/housing_regression_missing_values_missing_target.csv")
x_cols = [c for c in df.columns if c != "MEDV"]
X = df[x_cols]
y = df["MEDV"]
automl = AutoML(
algorithms=["CatBoost"],
mode="Compete",
optuna_time_budget=10,
eval_metric="r2",
kmeans_features=False,
golden_features=False,
mix_encoding=False,
random_state=44)
automl.fit(X, y)
print("*********************")
df["predictions"] = automl.predict(X)
print("Predictions")
print(df[["MEDV", "predictions"]].head())
The Decision Tree
model is present in the AutoML directory. It shouldn't. It was used for validation adjustment.
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working