我应该使用哪种TensorFlow来预测某人是否会喜欢某个地方？

Question

我是TensorFlow和机器学习的新手。我想训练TensorFlow来预测某人是否会使用他们喜欢的活动来喜欢某些地方。我应该使用哪种TensorFlow？

这是我的示例数据。

ID  Hiking  Swimming    Eating  Surfing  Mountain Climbing  Beach

1    -27       31         12      11         -66             1

2    -44       65        -29      87         -49             3

3     6       -82         78     -79          16            -5

4     52      -36        -84     -37          75            -3

5     -7       36         86      50         -83             5

注意：正值表示他们喜欢它。负值表示他们不喜欢它。

我希望返回一个类似的值： Beach = 3 如果我输入前5列的值，例如：

Hiking = -12
Swimming = 20
Eating = 10
Surfing = 50
Mountain Climbing = -40

Answer 1

如果您不熟悉深度学习，则可以尝试堆叠自己的DNN。这将有助于理解Tensorflow的工作原理。

示例代码： tf.estimator.DNNClassifier