"allowed families" argument ignores LightGBM about evalml HOT 2 CLOSED

Hi @ismorphism, thanks for filing. This is certainly a documentation failure on our part. The allowed_model_families argument only affects searches run with the IterativeAlgorithm, rather than our default DefaultAlgorithm. To limit to LightGBM, I would recommend this code instead:

automl = AutoMLSearch(
    X_train=X_train,
    y_train=y_train,
    data_splitter=cv,
    problem_type="binary",
    objective='f1',
    additional_objectives=["auc", "precision"],
    allowed_model_families=["lightgbm"],
    automl_algorithm="iterative",
    max_batches=3,
    optimize_thresholds=True,
    verbose=True,
)

We will update the documentation to include this.

With respect to allowed_model_families vs excluded_model_families, a quick look over shows that that was a mistake as well, exceluded_model_families should be able to handle both. We will fix!

from evalml.

@eccabay Thanks for the answer! 😊

from evalml.