gpt4 book ai didi

tensorflow - 在训练模型上调用预测时,Tensorflow 中的 KeyError

转载 作者:行者123 更新时间:2023-12-03 12:49:13 25 4
gpt4 key购买 nike

我已经训练了一个具有两个特征的 LinearRegressor:x,y 和标签: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)

训练后,我想从两个新值中进行预测
new_sample = np.array([20,20])
m.predict(new_sample)

但我在调用 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'

有谁知道我为什么会收到 KeyError?

最佳答案

尝试这个:

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)

关于tensorflow - 在训练模型上调用预测时,Tensorflow 中的 KeyError,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39687554/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com