Skip to content

Not deleteting validation adjustment model #353

@pplonski

Description

@pplonski

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 working

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions