Reputation: 47
Y_train.shape
Output is
(650,)
but I want output as
(650,1)
What should I do?
Upvotes: 0
Views: 27
Reputation: 1373
This should do the trick-
Y_train = Y_train[:, np.newaxis]
Upvotes: 1