CODEWITHSUNDEEP
tensorflow
Niclas
Niclas

Reputation: 156

KeyError in Tensorflow when calling predict on trained model

I have trained a LinearRegressor with two features: x,y and the label: l

def train_input_fn():
    x = [1,2,3,4]
    y = [2,3,4,5]
    feature_cols = tf.constant(x)
    labels = tf.constant(y)
    return feature_cols, labels    

x = tf.contrib.layers.real_valued_column("x")
y = tf.contrib.layers.real_valued_column("y")    
m = tf.contrib.learn.LinearRegressor(feature_columns=[ x,y],
                                      model_dir=model_dir)
m.fit(input_fn=train_input_fn, steps=100)

After training I want to predict from two new values

new_sample = np.array([20,20])
m.predict(new_sample)

but I get this error message when calling predict

File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/layers/python/layers/feature_column.py", line 870, in insert_transformed_feature
input_tensor = columns_to_tensors[self.name]
KeyError: 'x'

Does anyone know why I get KeyError?

Upvotes: 4

Views: 4821

Answers (2)

Abel Buriticá
Abel Buriticá

Reputation: 23

I am not an expert in Tensorflow but this works for me:

new_sample = np.array([20,20],dtype='float32')
empty_y = np.zeros(len(new_sample),dtype='float32')
prediction_x = tf.contrib.learn.io.numpy_input_fn({"x":new_sample},empty_y, batch_size=45,    num_epochs=100)
forecast = list(estimator.predict(input_fn=prediction_x,as_iterable=False))

Upvotes: 0

Kelly Stevens
Kelly Stevens

Reputation: 319

Try this:

my_feature_columns = [tf.contrib.layers.real_valued_column("", dimension=2)]
m = tf.contrib.learn.LinearRegressor(feature_columns=my_feature_columns,
                                     model_dir=model_dir)
m.fit(input_fn=train_input_fn, steps=100)

Upvotes: 1

Related Questions