feature\u列的Tensorflow 1.x项必须是\u FeatureColumn。给定类型为Lis

2024-06-28 10:50:10 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用TensorFlow1.15进行一个机器学习项目,在这个项目中,我使用tf.feature_column创建特性列,但是遇到了一个错误

下面是我如何定义feature columns

# Define feature columns

is_male = tf.feature_column.categorical_column_with_vocabulary_list('is_male', vocabulary_list=['False', 'True'])
plurality = tf.feature_column.categorical_column_with_vocabulary_list('plurality', vocabulary_list=['Single(1)', 'Twins(2)', 'Triplets(3)', 'Quadruplets(4)', 'Quintuplets(5)'])
wide_columns = [is_male, plurality]
mother_age = tf.feature_column.numeric_column(key="mother_age", shape=10)
gestation_weeks = tf.feature_column.numeric_column(key="gestation_weeks", shape=10)
deep_columns = [mother_age,gestation_weeks]
feature_cols = [wide_columns, deep_columns]

下面是我得到的确切错误:

ValueError: Items of feature_columns must be a _FeatureColumn. Given (type ): [VocabularyListCategoricalColumn(key='is_male', vocabulary_list=('False', 'True'), dtype=tf.string, default_value=-1, num_oov_buckets=0), VocabularyListCategoricalColumn(key='plurality', vocabulary_list=('Single(1)', 'Twins(2)', 'Triplets(3)', 'Quadruplets(4)', 'Quintuplets(5)'), dtype=tf.string, default_value=-1, num_oov_buckets=0)].


Tags: columns项目keyageistfcolumnmale