使用TensorFlow嵌入列时出现‘所有特征列必须是_FeatureColumn实例’错误求助
解决tf.feature_column使用时的ValueError问题
嘿,我一眼就揪出问题所在啦!你碰到的这个ValueError,核心原因是给columns变量多套了一层方括号,导致传入tfc.input_layer和tfc.make_parse_example_spec的是二维列表,而不是它们期望的一维FeatureColumn对象列表。
咱们来拆解下代码里的问题:
你写的代码片段里:
columns = [[tfc.embedding_column(colors, 3)], ...]
这里的[[...]]把embedding_column对象包裹进了子列表,变成了「列表的列表」。但tfc.input_layer和tfc.make_parse_example_spec都要求参数是直接包含FeatureColumn实例的一维列表,不能嵌套。当函数拿到嵌套列表时,会把里面的子列表判定为非FeatureColumn元素,自然就抛出错误了。
修正后的代码
把外层多余的方括号去掉,让columns直接是FeatureColumn对象的一维列表就好:
# 去掉外层[],改为一维列表 columns = [tfc.embedding_column(colors, 3), ...] features = tf.io.parse_example(..., features=tfc.make_parse_example_spec(columns)) dense_tensor = tfc.input_layer(features, columns)
修改后,columns里的每个元素都是标准的EmbeddingColumn实例,就能顺利通过函数的参数校验,正常运行啦。
内容的提问来源于stack exchange,提问作者Wha
相关产品推荐
相关产品推荐

