当前位置：网站首页>TF中的One-hot

TF中的One-hot

2022-08-11 06:45:00 【phac123】

tf.one_hot (待转换数据, depth=几分类)

import tensorflow as tf

classes = 3
labels = tf.constant([1, 0, 2])
output = tf.one_hot(labels, classes)
print(output)

在这里插入图片描述

注: 当列表中的变量带有小数的时候,one-hot报错

原网站

版权声明
本文为[phac123]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_42596275/article/details/126259236