print("----------------------------Datasets Loaded----------------------------") print("-----------------------Baseline Predictor Testing----------------------") space = (numpy.linspace(1, 300, 100)).astype(int) scores = [] model = Baseline() model.train(X, Y) for k in space: print("Epoch: ", k) model.set_beta_u(k) model.set_beta_i(k) score = model.RMSE(X_test, Y_test) print("RMSE: ", score) scores.append(score) matplotlib.pyplot.plot(space, scores, 'ro') matplotlib.pyplot.xlabel('Beta') matplotlib.pyplot.ylabel('RMSE') matplotlib.pyplot.title('Baseline Predictor') matplotlib.pyplot.savefig('../plots/baseline_predictor.png') matplotlib.pyplot.gcf().clear() print("------------------Baseline Predictor Testing Complete------------------") print("----------------User User Collaborative Filtering Testing--------------")