2017-01-09 22 views
0

私はthisチュートリアルの_tensorflow.orgに従っています。 input_fn_を正しく処理しようとしています。.fit()の引数として使用しています。 私は分類器作成しました:入力関数input_fn()の作成方法

classifier = tf.contrib.learn.SKCompat(tf.contrib.learn.DNNClassifier(
feature_columns=feature_cols, 
hidden_units=[10, 10], 
model_dir=("C:\\........\tmp"), 
n_classes=2, 
activation_fn=tf.sigmoid, 
optimizer=tf.train.ProximalAdagradOptimizer(
    learning_rate=0.1, 
    l1_regularization_strength=0.001 
    ))) 

を次に入力機能:

def input_fn(data_set): 
    feature_cols = {k: tf.constant(data_set[k].values) 
        for k in FEATURES} 
    labels = tf.constant(data_set[LABEL].values) 
    return feature_cols, labels 

最後に私が入れているinput_fn()フィット(中)によって:

classifier.fit(input_fn=lambda: input_fn(training_set), steps=10) 

コードを実行すると、次のエラーが表示されます。

TypeError         Traceback (most recent call last) 
<ipython-input-6-938bcd2f929f> in <module>() 
----> 1 classifier.fit(input_fn=lambda: input_fn(training_set), steps=10) 

TypeError: fit() got an unexpected keyword argument 'input_fn' 

それは input_fn定義についてであれば、私は知らない、または引数

に合わせ

答えて

0

あなたは、input_fn使用して最初の行を置き換えたい場合はSKCompatを使用しないでください:

classifier = tf.contrib.learn.DNNClassifier(

必要に応じてかっこを調整します。

関連する問題