9fd84ad62217e80e521f755c151a2d1dbc9ebe2d,autokeras/search.py,Searcher,load_best_model,#Searcher#,55

Before Change

    def load_best_model(self):
        return model with best accuracy
        model_id = max(self.history, key=lambda x: x["accuracy"])["model_id"]
        return load_model(os.path.join(self.path, str(model_id) + ".h5"))

    def add_model(self, model, x_train, y_train, x_test, y_test):
        add one model while will be trained to history list

After Change

    def load_best_model(self):
        return model with best accuracy
        return self.load_model_by_id(max(self.history, key=lambda x: x["accuracy"])["model_id"])

    def add_model(self, model, x_train, y_train, x_test, y_test):
        add one model while will be trained to history list

Instances