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

View file

@ -404,7 +404,7 @@ def robot_wireless(data_set='robot_wireless'):
Y = allY[0:215, :]
Xtest = allX[215:, :]
Ytest = allX[215:, :]
Ytest = allY[215:, :]
return data_details_return({'X': X, 'Y': Y, 'Xtest': Xtest, 'Ytest': Ytest, 'addresses' : addresses, 'times' : times}, data_set)
def silhouette(data_set='ankur_pose_data'):