Как конвертировать размеры массива numpy?

Question

Y_train.shape

Вывод

(650,)

, но я хочу вывод как

(650,1)

Что мне делать?

Answer 1

Это должно сработать-

Y_train = Y_train[:, np.newaxis]