Как реализовать одно горячее кодирование без каких-либо кадров ML в python? - PullRequest
0 голосов
/ 29 апреля 2020

делает учебник MNIST без MLframe, но застрял на одном этапе горячего кодирования

y содержит данные меток цифровых изображений

0, 1, 2, 3, 4, 5, 6 , 7, 8, 9

и его размер (10000,)

Я хочу преобразовать номера каждой категории в один массив горячего кодирования

0: 1 0 0 0 0 0 0 0 0 0 0
1: 0 1 0 0 0 0 0 0 0 0 0
2: 0 0 1 0 0 0 0 0 0 0 0 и т. Д.

, поэтому я сделал код

import numpy as np
y_one=np.zeros(y.size, 10)
y_one[np.arange(y.size), y]=1

он говорит «тип данных не понят»

Как реализовать одно горячее кодирование без sklearn или tf в этом случае?

1 Ответ

0 голосов
/ 29 апреля 2020

Передать кортеж целого числа для shape параметра np.zeros функции

y_one = np.zeros((y.size, 10))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...