neural-network - TensorFlowDNNClassifier 或 TensorFlow 中的任何其他深度神经网络模型是否支持浮点值的预测？

Question

在 TensorFlowDNNClassifier 中，我必须指定 n_classes（类数）。如果我预测一个介于 0 和 1 之间的浮点值怎么办？如果我想要完全准确，可能有无限数量的类。

更清楚地说，TensorFlow skflow 中的神经网络是为分类问题而设计的。但是，我正在处理一个价值预测问题。如何针对这种浮点值预测问题修改模型？

Answer 1

预测 0 到 1 之间的值的任务称为回归。相应地，您需要 aTensorFlowDNNRegressor而不是TensorFlowDNNClassifier.

接口的唯一区别是，当您调用 时fit，y不是类标签，而是 0 和 1 之间的值。同样，predict返回的值也是 0 和 1 之间的值。