Tensorflow功能列为0和1

时间:2017-12-05 04:34:48

标签: python-3.x tensorflow

对于Tensorflow,要素列包含布尔值0和1。

我应该使用tf.feature_column.numeric_column

吗?

tf.feature_column.categorical_column_with_vocabulary_list

2 个答案:

答案 0 :(得分:0)

我会推荐tf.feature_column.categorical_column_with_vocabulary_list。

我认为您希望将数据视为类别,而不是标量。

答案 1 :(得分:0)

您可以使用sequence_categorical_column_with_identity:

tf.feature_column.sequence_categorical_column_with_identity('YOUR_FEATURE',2)

第二个参数是存储桶数。如果您的特征中有n个连续的整数[0,1,2 ..,n-1],则应输入:

tf.feature_column.sequence_categorical_column_with_identity('YOUR_FEATURE',n)