TensorFlow Estimator的功能列和标签列

时间:2019-08-14 06:03:18

标签: tensorflow-estimator

我试图理解TensorFlow Estimator API的概念。据我了解,必须为估计器(例如LinearRegressor)指定特征列,以便将数据打包到LinearRegressor期望的输入向量中。如果是这样,为什么我们不需要指定输出列呢?

例如,在下面的示例中,我们训练了LinearRegressor以预测房地产价格。输入列是平方英尺和财产类型(房屋或公寓)。标签列(价格)根本没有传递到LinearRegressor。据我了解,标签取自train_input_fn,但要素列也可以取自标签,但我们仍然传递featcols而不是标签。

enter image description here

0 个答案:

没有答案