如何快速生成多类别的one-hot向量
Keras
keras 的版本是 2.3.1
>>> import numpy as np
>>> import keras
>>> labels = np.random.randint(10, size=(10, 1))
>>> labels
array([[7],
[1],
[8],
[9],
[3],
[1],
[1],
[6],
[1],
[8]])
>>> one_hot_labels = keras.utils.to_categorical(labels, num_classes=10)
>>> one_hot_labels
array([[0., 0., 0., 0., 0., 0., 0., 1., 0., 0.],
[0., 1., 0., 0., 0.
共有 0 条评论