else:
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------------------")
