生成one-hot编码

import numpy as np
from sklearn import preprocessing

print '-----------第1种方法：使用python库--------------------------------------------------'
batch_y = np.array([[0],[1],[5]])      #标签
enc = preprocessing.OneHotEncoder()
microvideolabel = np.array([[0], [1], [2], [3],[4], [5]     
              ])
enc.fit(microvideolabel)
batch_y = enc.transform(batch_y).toarray()
print batch_y

print '-----------第2种方法：自己实现--------------------------------------------------'
batch_y = np.array([[0],[1],[5]])     #标签
num_labels = batch_y.shape[0]  #获取标签的个数
 
index_offset = np.arange(num_labels) * 6     # 每个标签的结束位置
 
labels_one_hot = np.zeros((num_labels, 6))   # 每个类生成6个0

#batch_y.ravel()   # 散开，解开 由矩阵变成向量 [0 1 5]
labels_one_hot.flat[index_offset + batch_y.ravel()] = 1

print labels_one_hot

y_onehot = tf.one_hot(y, 10)