Minor fixes to regression example with robot_wireless.

This commit is contained in:
Neil Lawrence 2013-08-19 00:04:46 +02:00
parent 4082f6c02e
commit 2004cf3ea9
2 changed files with 5 additions and 5 deletions

View file

@ -334,16 +334,16 @@ def robot_wireless(optim_iters=100):
# optimize
m.optimize(messages=True, max_f_eval=optim_iters)
Ypredict = m.predict(data['Y'])[0]
Xpredict = m.predict(data['Ytest'])[0]
pb.plot(data['Xtest'][:, 0], data['Xtest'][:, 1], 'r-')
pb.plot(Ypredict[:, 0], Ypredict[:, 1], 'b-')
pb.plot(Xpredict[:, 0], Xpredict[:, 1], 'b-')
pb.axis('equal')
pb.title('WiFi Localization with Gaussian Processes')
pb.legend(('True Location', 'Predicted Location'))
sse = ((data['Ytest'] - Y.predict)**2).sum()
sse = ((data['Xtest'] - Xpredict)**2).sum()
print(m)
print('Sum of squares error on test data: ', str(sse))
print('Sum of squares error on test data: ' + str(sse))
return m
def sparse_GP_regression_1D(N=400, num_inducing=5, optim_iters=100):