python-3.x - tensorflow.contrib.learn.DNNclassifier 缺少 1 个必需的位置参数 : 'feature_columns'

Question

我正在使用 jupyter 笔记本编写深度神经网络代码。我在尝试创建 DNNClassifier 时遇到了这个问题。

import tensorflow.contrib.learn as learn
classifier = learn.DNNClassifier(hidden_units=[10, 20, 10], n_classes=2)

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-30-0273d6847e2b> in <module>()
----> 1 classifier = learn.DNNClassifier(hidden_units=[10, 20, 10], n_classes=2)

TypeError: __init__() missing 1 required positional argument: 'feature_columns'

Answer 1

您需要传递正确的参数tf.contrib.learn.DNNClassifier ，这里您没有传递feature_columns参数。

例如，您可以使用real_valued_column作为features_columns

feature_columns = [tf.contrib.layers.real_valued_column("", dimension=1)]
classifier = learn.DNNClassifier(feature_columns=feature_columns, hidden_units=[10, 20, 10], n_classes=2)