Skip to content

Updated excluded_model_families data type - #4287

Merged
remyogasawara merged 9 commits into
mainfrom
4259_update_exluded_model_families_doc
Aug 29, 2023
Merged

remyogasawara merged 9 commits into
mainfrom
4259_update_exluded_model_families_doc

Conversation

@remyogasawara

@remyogasawara remyogasawara commented Aug 24, 2023

Copy link
Copy Markdown
Contributor

Resolves #4259

Updated excluded_model_families to allow both str and ModelFamily data types and updated corresponding documentation.

@remyogasawara remyogasawara self-assigned this Aug 24, 2023
@codecov

codecov Bot commented Aug 24, 2023

Copy link
Copy Markdown

Codecov Report

Patch coverage: 100.0% and project coverage change: +0.1% 🎉

Comparison is base (ecbb2e4) 99.7% compared to head (2ccf674) 99.7%.

Additional details and impacted files
@@           Coverage Diff           @@
##            main   #4287     +/-   ##
=======================================
+ Coverage   99.7%   99.7%   +0.1%     
=======================================
  Files        355     355             
  Lines      39154   39155      +1     
=======================================
+ Hits       39034   39035      +1     
  Misses       120     120             
Files Changed Coverage Δ
...valml/automl/automl_algorithm/default_algorithm.py 99.7% <ø> (ø)
...lml/automl/automl_algorithm/iterative_algorithm.py 97.5% <ø> (ø)
evalml/pipelines/components/utils.py 96.8% <ø> (ø)
evalml/automl/automl_search.py 99.8% <100.0%> (ø)
evalml/tests/automl_tests/test_automl.py 99.5% <100.0%> (+0.1%) ⬆️

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

problem_type=ProblemTypes.BINARY,
automl_algorithm=automl_algorithm,
excluded_model_families=[ModelFamily.RANDOM_FOREST],
excluded_model_families=[ModelFamily.RANDOM_FOREST, "linear_model"],

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let's also assert than any linear models are not created as well.

@eccabay eccabay left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you!

@remyogasawara
remyogasawara merged commit 032db03 into main Aug 29, 2023
@remyogasawara
remyogasawara deleted the 4259_update_exluded_model_families_doc branch August 29, 2023 19:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

"allowed families" argument ignores LightGBM

3 participants