Feature columns tensorflow
numerical_columns=['Length','Diameter','Height','Whole_weight','Shucked_weight','Viscera_weight','Shell_weight'] categorical_columns=['Sex'] feature_columns=[] for value in numerical_columns: feature_columns.append(tf.feature_column.numeric_column(value,dtype=tf.float32)) for value in categorical_columns: unique_values=dataset[value].unique() feature_columns.append(tf.feature_column.categorical_column_with_vocabulary_list(value,unique_values)) feature_columns