a8eabdad14eee8a47257248fa271700fcce939cb,autokeras/search.py,BayesianSearcher,search,#BayesianSearcher#Any#Any#Any#Any#,159

Before Change

            self.search_tree.add_child(-1, history_item["model_id"])
            self.gpr.first_fit(Graph(model).extract_descriptor(), history_item["accuracy"])
            pickle_to_file(self, os.path.join(self.path, "searcher"))
            del model
            backend.clear_session()

        else:
            model_ids = self.search_tree.get_leaves()

After Change

            history_item = self.add_model(model, x_train, y_train, x_test, y_test, constant.SEARCH_MAX_ITER)
            self.init_gpr_x.append(graph.extract_descriptor())
            self.init_gpr_y.append(history_item["accuracy"])
            pickle_to_file(self, os.path.join(self.path, "searcher"))
            return

        if not self.init_search_queue and not self.gpr.first_fitted:

Instances