[tune] Tune experiment analysis improvements (#10645)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Kai Fricke
2020-09-08 21:00:52 -07:00
committed by GitHub
co-authored by Richard Liaw
parent d9c68fca5c
commit d7c7aba99c
14 changed files with 247 additions and 38 deletions
+4 -2
View File
@@ -285,8 +285,10 @@ class BayesOptSearch(Searcher):
analysis (ExperimentAnalysis): Optionally, the previous analysis
to integrate.
"""
for (_, report), params in zip(analysis.dataframe().iterrows(),
analysis.get_all_configs().values()):
for (_, report), params in zip(
analysis.dataframe(metric=self._metric,
mode=self._mode).iterrows(),
analysis.get_all_configs().values()):
# We add the obtained results to the
# gaussian process optimizer
self._register_result(params, report)