label one-hot编码
发现from keras.utils.np_utils import to_categorical 的to_categorical函数和pandas.get_dummies()实现的功能相同。
都是对目标对象 one-hot编码
- to_categorical(y, num_classes=None, dtype=‘float32’)
- def get_dummies(
data,
prefix=None,
prefix_sep="_",
dummy_na=False,
columns=None,
sparse=False,
drop_first=False,
dtype=None,
)
print(type(pd.get_dummies(Y_train).values))
print(to_categorical(Y_train, num_classes = 10)==pd.get_dummies(Y_train).values)
print(type(to_categorical(Y_train, num_classes = 10)))
<class ‘numpy.ndarray’>
[[ True True True … True True True]
[ True True True … True True True]
[ True True True … True True True]
…
[ True True True … True True True]
[ True True True … True True True]
[ True True True … True True True]]
<class ‘numpy.ndarray’>