[英]ValueError: features should be a dictionary of `Tensor`s. Given type: <class 'tensorflow.python.framework.ops.Tensor'>
[英]TensorFlow - ValueError: features should be a dictionary of `Tensor`s
我正在尝试使用TensorFlow编写一个模型,该模型利用预制的DNNClassifier来对鸢尾花进行分类。
BATCH_SIZE=100
TRAIN_STEPS=1000
def main(unused_argv):
# Load training and eval data
iris = tf.contrib.learn.datasets.load_dataset("iris")
data = iris.data
labels = iris.target
train_data = data[:100]
train_labels = labels[:100]
eval_data = data[100:]
eval_labels = labels[100:]
# Feature columns
feature_columns = [
tf.feature_column.numeric_column(key="SepalLength"),
tf.feature_column.numeric_column(key="SepalWidth"),
tf.feature_column.numeric_column(key="PetalLength"),
tf.feature_column.numeric_column(key="PetalWidth")
]
# Build DNN with 2 hidden layers, 10 neurons each
classifier = tf.estimator.DNNClassifier(feature_columns=feature_columns, hidden_units=[10, 10], n_classes=3)
# Train model
train_input_fn = tf.estimator.inputs.numpy_input_fn(
x=train_data,
y=train_labels,
batch_size=BATCH_SIZE,
num_epochs=None,
shuffle=True
)
classifier.train(input_fn=train_input_fn, steps=TRAIN_STEPS)
# Evaluate model
eval_input_fn = tf.estimator.inputs.numpy_input_fn(
x=eval_data,
y=eval_labels,
num_epochs=1,
shuffle=False
)
eval_results = classifier.evaluate(input_fn=eval_input_fn)
print(eval_results)
我收到以下错误:
ValueError:功能应该是
Tensor
的字典。
据我了解,功能列应该是列表。 我的实现有什么问题?
你的tensorflow版本是什么?
根据这个 ,如果你使用1.4或降低可以通过升级来解决这个错误。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.