import numpy as np
from sklearn.preprocessing import LabelBinarizer
def One_hot(labels):
num_class=10
lb=LabelBinarizer().fit(np.array(range(num_class)))
labels=lb.transform(labels)
return labels
深度学习——对标签进行One-hot编码
最新推荐文章于 2024-10-07 23:21:25 发布