如何预测张量流中只有一个类

Question

如果您只想预测一个班级。然后，首先需要以这样的方式标记您的向量，即可以将所有这些向量标记为“一个”。它具有基础事实5和“零”＆＃39;那些基本事实不是5的载体。 如何使用puthon

在tensorflow中实现这一点

Answer 1

在准备数据时，您可以使用numpy将第5类中的所有数据点设置为1，而其他数字将使用设置为0。

arr = np.where(arr!=5,arr,0)

arr = np.where(arr=5,arr,1)

然后您可以使用Tensorflow创建二进制分类器，同时使用binary_crossentropy损失对分类器进行优化