def fitness(self, nn_chromosome: BackPropModel, train_dataset, val_dataset): print("train") nn_chromosome.train(train_dataset) print("test") accuracy = nn_chromosome.test(val_dataset) print(accuracy) return accuracy
def train_best(self, best: BackPropModel, dataset): best.args.epochs = 1 best.train(dataset) best.args.epochs = 1