Я работаю над проектом машинного обучения с использованием TensorFlow 1.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: Элементы feature_columns должны быть _FeatureColumn. Дано (тип): [VocabularyListCategoricalColumn (key = 'is_male', vocabulary_list = ('False', 'True'), dtype = tf.string, default_value = -1, num_oov_buckets = 0), VocabularyListCategoricalColumn (key = 'множественность'vocabulary_list = ('Single (1)', 'Twins (2)', 'Triplets (3)', 'Quadruplets (4)', 'Quintuplets (5)'), dtype = tf.string, default_value = -1,num_oov_buckets = 0)].