1. 程式人生 > >tensorflow相關函式__ to_categorical

tensorflow相關函式__ to_categorical

函式宣告: to_categorical(y, num_classes=None, dtype='float32')

作用:將整型標籤轉為onehot。y為int陣列,num_classes為標籤類別總數,大於max(y)(標籤從0開始的)。

返回:如果num_classes=None,返回len(y) * [max(y)+1](維度,m*n表示m行n列矩陣,下同),否則為len(y) * num_classes。
 

例子:
 

import keras

ohl=keras.utils.to_categorical([1,3])
# ohl=keras.utils.to_categorical([[1],[3]])
print(ohl)
"""
[[0. 1. 0. 0.]
 [0. 0. 0. 1.]]
"""
ohl=keras.utils.to_categorical([1,3],num_classes=5)
print(ohl)
"""
[[0. 1. 0. 0. 0.]
 [0. 0. 0. 1. 0.]]
"""

簡而言之,就是將給定的序列,編碼成一個one -hot矩陣。注意編號是從0開始的。

注:import keras 也可以通過:import tensorflow.contrib.keras as kr 來實現匯入keras