-
Notifications
You must be signed in to change notification settings - Fork 491
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add warmstart test #298
add warmstart test #298
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Use the starting point as init_value in case one forgets to pass starting_points. Not necessary in this example.
test/automl/test_warmstart.py
Outdated
hps_to_search = { | ||
"n_estimators": { | ||
"domain": tune.lograndint(lower=10, upper=32768), | ||
"init_value": 32768, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"init_value": 32768, | |
"init_value": space["n_estimators"]["domain"], |
test/automl/test_warmstart.py
Outdated
}, | ||
"num_leaves": { | ||
"domain": tune.lograndint(lower=10, upper=3276), | ||
"init_value": 3276, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"init_value": 3276, | |
"init_value": space["num_leaves"]["domain"], |
test/automl/test_warmstart.py
Outdated
def test_fit_w_freezinghp_starting_point(self, as_frame=True): | ||
automl_experiment = AutoML() | ||
automl_settings = { | ||
"time_budget": 3, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"time_budget": 3, | |
"time_budget": 1, |
) | ||
starting_points[new_estimator_name] = starting_points["lgbm"] | ||
automl_settings_resume = { | ||
"time_budget": 3, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"time_budget": 3, | |
"time_budget": 1, |
No description provided.