我想将一个包含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")