Tensorflow：为特征变量创建循环

Question

我在TensorFlow中有多个功能列，我正在尝试创建循环以避免手动键入以创建功能列，但它无法正常工作。下面是我创建要素列所需的所有列的列表（这只是一个虚拟数据，但如果我们有数百列，那该怎么办）。

num_preg = tf.feature_column.numeric_column('Number_pregnant')
glucose_conc = tf.feature_column.numeric_column('glucose_concentration')
blood_prs = tf.feature_column.numeric_column('blood_pressure')
Tricep = tf.feature_column.numeric_column('Triceps')
insulin = tf.feature_column.numeric_column('Insulin')

I created a loop for this as mentioned below, df_col_num is a list containing the column names for which I need to create feature column.

for col in df_col_num:
    col= tf.feature_column.categorical_column_with_hash_bucket(col,hash_bucke[enter link description here][1]t_size=50)

任何帮助或建议都将不胜感激。

谢谢！

Answer 1

您的变量是数字，但您使用的是tf.feature_column.categorical_column_with_hash_bucket，这是分类变量。

我会用：

variable_names = df.columns

tf_variables =[]

for index in variable_names:
    tf_variables.append(tf.feature_column.numeric_column(index)) 

这对我有用。