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: (1, 2000) x_test: (1000, 2) y_test: (1, 1000)