Exemplo n.º 1
0
matplotlib.pyplot.gcf().clear()

print("------------------Baseline Predictor Testing Complete------------------")


print("----------------User User Collaborative Filtering Testing--------------")

model = UserUserCollaborativeFiltering()
model.train(X, Y)
space = (numpy.linspace(1, 2000, 100)).astype(int)
scores = []
times = []

for k in space:
	print("Epoch: %i", k)
	model.set_neighbourhood(k)
	if k > 1:
		start_time = time.time()
	score = model.RMSE(X_test, Y_test)
	if k > 1:
		times.append(time.time() - start_time)
		print("Time: ", time.time() - start_time)
	print("RMSE: ", score)
	scores.append(score)

matplotlib.pyplot.plot(space, scores, '+')
matplotlib.pyplot.xlabel('Nearest Neighbours')
matplotlib.pyplot.ylabel('RMSE')
matplotlib.pyplot.title('User - User Collaborative Filtering')
matplotlib.pyplot.savefig('../plots/user_user_collaborative_filtering1.png')
matplotlib.pyplot.gcf().clear()