def main(): seed(6) fileName = 'database.txt' p = Problem(fileName) n_folds = 5 l_rate = 0.01 n_epoch = 5 p.prepareDataSet() dataset = p.getDataSet() c = Controller(p) sol = c.solveGradientDescent(dataset, c.regressionGradientDescent, n_folds, l_rate, n_epoch) scores = sol[0] print('Error: %.5f' % sol[1])