将34个特征输入tf.contrib.learn.DNNClassifier并按2个类获取预测

时间:2019-06-18 14:30:56

标签: python-3.x tensorflow

我想将一个包含DNN个功能列的csv文件提供给34,并且希望通过DNNClassifier中的两个类进行预测。

我尝试了这段代码,但无法获得模型。

feature_columns = [tf.contrib.layers.real_valued_column("", dimension=1)]

classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
                                            hidden_units=[35, 50, 2],
                                            n_classes=2,
                                            model_dir="/tmp1/gender_model")

0 个答案:

没有答案