wide_deep在我数据报错
来源:10-3 wide and deep网络结构与数学原理介绍

慕哥0458965
2019-01-27
老师,您好,我尝试在我的数据集上生成pb文件,但是模仿您的demo,生成pb文件的时候报错,请问您怎么解决?
报错内容:ValueError: feature_columns contain different parse_spec for key 20. Given VarLenFeature(dtype=tf.string) and VarLenFeature(dtype=tf.int64)
我的模仿代码如下:
else:
model_es=tf.estimator.DNNLinearCombinedClassifier(
model_dir=model_dir,
linear_feature_columns=wide_columns,
dnn_feature_columns=deep_columns,
dnn_hidden_units=hidden_units,
config=run_config)
feature_column = wide_columns + deep_columns
feature_spec = tf.feature_column.make_parse_example_spec(feature_column)
serving_input_fn = (tf.estimator.export.build_parsing_serving_input_receiver_fn(feature_spec))
model_es.export_savedmodel(’./model_pb/’, serving_input_fn)
非常期待您的回答!
1回答
-
David
2019-01-27
这里错误应该是第20个特征, 你数据传入有的是string 有的是int。看起来你数据预处理的应该有点问题。另外 你如果是自己的数据集 需要把feature transformation 改好。就是wide colums ,deep columns那里和特征名字对应好。
012019-01-28
相似问题