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

Question

老师您好：

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

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

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

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

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

请问该怎么处理嗯

Answer 1

正十七 2019-10-17 23:03:35

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

https://www.tensorflow.org/api_docs/python/tf/feature_column/categorical_column_with_vocabulary_list

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

https://www.tensorflow.org/api_docs/python/tf/feature_column/crossed_column