evalml icon indicating copy to clipboard operation
evalml copied to clipboard

allowed_model_families is not enforced

Open iXanthos opened this issue 1 year ago • 2 comments

I want evalml to only create XGBOOST models in the allotted time, but both baseline and Random Forest models are also generated. My hyperparameters and my setup are:

Code Sample, a copy-pastable example to reproduce your bug.

eval_automl = automl.AutoMLSearch(
        X_train=train_data,
        y_train=train_target,
        problem_type="binary",
        max_time=60,
        allowed_model_families=["xgboost"],
        ensembling=False,
        n_jobs=1,
        verbose=True,
    )

I understand the baseline could be mandatory to ensure a result, but why is random forests also included?

iXanthos avatar Nov 26 '24 08:11 iXanthos

hello, any response on this?

iXanthos avatar Dec 09 '24 08:12 iXanthos

Hello, I'd like to join the project by working on this. Could you assign me the issue? Thank you

Riccardo231 avatar Dec 26 '24 19:12 Riccardo231