请问有一个csv的字段是变长的列表怎么输入到feature column到模型

来源：2-13 子类API实现wide&deep模型

蚂蚁帅帅

2019-10-16

老师您好：

我有一个csv，有个字段是，分类列表的json

比如[“分类1”, “分类2”, “分类3”]
请问怎么处理才可以输入到模型？

我可以使用
tf.feature_column.categorical_column_with_vocabulary_list
加上
tf.feature_column.indicator_column

然而我直接把csv加载到pandas，然后把这一列json.loads，得到一个变长了列表

tensorflow报错说，这个shape不固定不能是变长列表

请问该怎么处理嗯

写回答

1回答

正十七

2019-10-17

已采纳

对，这个categorical_column_with_vocabulary_list 的输入key不能是列表：

这个时候，如果你想要把变长列表输入进来，建议每一个都用这个方法做成一个独立的feature，如果他们之间有关联关系的话，可以考虑使用cross_column来做交叉特征。

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 1

回答 2

回答 1

回答 1

回答 1