Tensorflow 分类预测

如何解决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])

谢谢。