Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
bada5ca
Initial commit
ParthivNaresh May 13, 2021
cf9009a
update release notes
ParthivNaresh May 13, 2021
80f6826
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 13, 2021
6a4821c
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 13, 2021
e3dac68
wording change
ParthivNaresh May 14, 2021
4f6812a
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 14, 2021
3767c18
Update test
ParthivNaresh May 15, 2021
e5484f3
update tests
ParthivNaresh May 15, 2021
7e11eaa
dependency update
ParthivNaresh May 15, 2021
71ffe5c
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 15, 2021
2872eeb
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 15, 2021
faad1b6
Merge branch 'Update-Logging-Information-AutoML-Init' of ssh://github…
ParthivNaresh May 15, 2021
6fe642b
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 17, 2021
1e30ba2
simplify test
ParthivNaresh May 17, 2021
cca928d
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 17, 2021
7f139e9
change model families away from minimal dependencoes
ParthivNaresh May 17, 2021
85beee6
Merge branch 'Update-Logging-Information-AutoML-Init' of ssh://github…
ParthivNaresh May 17, 2021
f28275b
update test
ParthivNaresh May 18, 2021
280d0af
update test
ParthivNaresh May 18, 2021
ad5b49e
Merge branch 'main' into Update-Logging-Information-AutoML-Init
ParthivNaresh May 18, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/source/release_notes.rst
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ Release Notes
* Set minimum required version for for pyzmq, colorama, and docutils :pr:`2254`
* Changed BaseSampler to return None instead of y :pr:`2272`
* Changes
* Updated logging information in ``AutoMLSearch.__init__`` to clarify pipeline generation :pr:`2263`
* Removed ensemble split and indices in ``AutoMLSearch`` :pr:`2260`
* Updated pipeline ``repr()`` and ``generate_pipeline_code`` to return pipeline instances without generating custom pipeline class :pr:`2227`
* Documentation Changes
Expand Down
2 changes: 2 additions & 0 deletions evalml/automl/automl_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -380,6 +380,8 @@ def __init__(self,

if self.allowed_pipelines == []:
raise ValueError("No allowed pipelines to search")

logger.info(f"{len(self.allowed_pipelines)} pipelines ready for search.")
check_all_pipeline_names_unique(self.allowed_pipelines)

run_ensembling = self.ensembling
Expand Down
22 changes: 22 additions & 0 deletions evalml/tests/utils_tests/test_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

import pytest

from evalml import AutoMLSearch
from evalml.utils.logger import (
get_logger,
log_subtitle,
Expand Down Expand Up @@ -210,3 +211,24 @@ def test_time_elapsed(mock_time, time_passed, answer):
mock_time.return_value = time_passed
time = time_elapsed(start_time=0)
assert time == answer


@pytest.mark.parametrize("type_, allowed_families, number_, number_min_dep", [("binary", None, 8, 5), ("multiclass", 2, 2, 2), ("regression", 3, 3, 3)])
def test_pipeline_count(type_, allowed_families, number_, number_min_dep, X_y_binary, X_y_multi, X_y_regression, caplog, has_minimal_dependencies):
if type_ == 'binary':
X, y = X_y_binary
elif type_ == 'multiclass':
X, y = X_y_multi
else:
X, y = X_y_regression
if not allowed_families:
_ = AutoMLSearch(X_train=X, y_train=y, problem_type=type_)
else:
if allowed_families == 2:
_ = AutoMLSearch(X_train=X, y_train=y, problem_type=type_, allowed_model_families=['random_forest', 'decision_tree'])
elif allowed_families == 3:
_ = AutoMLSearch(X_train=X, y_train=y, problem_type=type_, allowed_model_families=['random_forest', 'decision_tree', 'extra_trees'])
if has_minimal_dependencies:
assert f"{number_min_dep} pipelines ready for search" in caplog.text
else:
assert f"{number_} pipelines ready for search" in caplog.text