2016-07-29 16 views
3

私はTensorflowで線形回帰を訓練しようとしています。私は自動的に実数値列を決定することによって、学習者をインスタンス化した場合列ヘッダーが存在するにもかかわらずKeyError?

フィッティングが正常に動作します。私は手動で列を指定しようとすると、

auto_feature_columns = 
    tf.contrib.learn.infer_real_valued_columns_from_input(
     training_examples) 
linear_regressor = tf.contrib.learn.LinearRegressor(
    feature_columns=auto_feature_columns, 
    optimizer=tf.GradientDescentOptimizer(learning_rate=0.002), 
    gradient_clip_norm=3.14 
) 

はしかし、私は(すべて3列用)KeyErrorsを取得する:私はそれらの3は私のトレーニングデータの特徴であることを確信していますので、奇数である

linear_regressor = tf.contrib.learn.LinearRegressor(
    feature_columns=[ 
     tf.contrib.layers.real_valued_column('tailLength'), 
     tf.contrib.layers.real_valued_column('whiskerLength'), 
     tf.contrib.layers.real_valued_column('playfulness') 
    ], 
    optimizer=tf.GradientDescentOptimizer(learning_rate=0.002), 
    gradient_clip_norm=3.14 
) 

。あなたは推定に合うようにしようとすると

+2

あなたはこれを理解しましたか? – Jordan

答えて

0

KeyError例外が起こるのでしょうか?

もしそうであれば、estimator.fitを使用する場合https://www.tensorflow.org/tutorials/input_fn/

に私は同じエラーを持っていたとしておそらく、それらのキーを有するfeature_columnsとラベルを返す機能操作を可能にする入力機能を使用しなければならない(X、Y )...

乾杯、

ラファエル。

関連する問題