12fac3a0ecc05215985c6b42b39447a3f2e04c0c,autokeras/search.py,BayesianSearcher,search,#BayesianSearcher#,163

Before Change

            history_item = self.add_model(model, x_train, y_train, x_test, y_test)
            self.search_tree.add_child(-1, history_item["model_id"])
            self.gpr.first_fit(Graph(model).extract_descriptor(), history_item["accuracy"])
            pickle.dump(self, open(os.path.join(self.path, "searcher"), "wb"))
            del model
            backend.clear_session()

        else:

After Change

        self.gpr = IncrementalGaussianProcess()
        self.search_tree = SearchTree()

    def search(self, x_train, y_train, x_test, y_test):
        if not self.history:
            model = DefaultClassifierGenerator(self.n_classes, self.input_shape).generate()
            history_item = self.add_model(model, x_train, y_train, x_test, y_test)
            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:

Instances