import tensorflow as tf
from tensorflow import keras
def props_to_onehot(props):
# props为概率输出
indices = np.argmax(props,axis=-1)
categorical_labels = keras.utils.to_categori
sortmax 结果转 onehot (tensorflow2.0)
最新推荐文章于 2022-04-03 19:13:18 发布
本文介绍了一种将概率输出转换为One-hot编码的方法。通过使用numpy.argmax()找到最大概率的索引,并利用keras.utils.to_categorical()将其转换为One-hot格式。此方法在深度学习和机器学习中用于将类别概率分布转换为可用于模型训练的目标变量。

最低0.47元/天 解锁文章
7465

被折叠的 条评论
为什么被折叠?



