Tensorflow 分类预测

问题描述

我正在处理一个简单的分类问题,而且我是新手。我希望它给出 0 和 1 之类的结果,但它给出的百分比结尾如下。我该如何解决这个问题?数据集中的标签列有2个目标。

input_dict = {name: tf.convert_to_tensor([value]) for name,value in sample.items()}
predictions = model.predict(input_dict)
prob= tf.nn.sigmoid(predictions[0])

谢谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)