Now we need to get a feel of the size of our dataset

In [6]:
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:	(2000, 2)
y_train:	(2000, 1)
x_test:		(1000, 2)
y_test:		(1000, 1)