To get a better sense of the data, let's checkout the shapes of the loaded arrays.

print('x_train:\t{}'.format(X_train.shape))
print('y_train:\t{}'.format(y_train.shape))
print('x_test:\t\t{}'.format(X_test.shape))
print('y_test:\t\t{}'.format(y_test.shape))

x_train:	(800, 2)
y_train:	(800, 1)
x_test:		(200, 2)
y_test:		(200, 1)