import tensorflow as tf from tensorflow.python.estimator.inputs import numpy_io import numpy as np import collections from tensorflow.python.framework import errors from tensorflow.python.platform import test from tensorflow.python.training import coordinator from tensorflow import feature_column from tensorflow.python.feature_column.feature_column import _LazyBuilder def test_numeric(): ''' 两个产生的结果相同 _LazyBuilder input_layer :return: ''' price = {'price': [[1.], [2.], [3.], [4.],[10.]]} # 4行样本 builder = _LazyBuilder(price) def transform_fn(x): return x + 2 price_column = feature_column.numeric_column('price', normalizer_fn=transform_fn) price_transformed_tensor = price_column._get_dense_tensor(builder) with tf.Session() as session: print(session.run([price_transformed_tensor])) # 使用input_layer price_transformed_tensor = feature_column.input_layer(price, [price_column]) with tf.Session() as session: print('use input_layer' + '_' * 40) print(session.run([price_transformed_tensor])) def test_bucketized_column(): price = {'price': [[5.], [15.], [25.], [35.], [55.], [45.]]} # 4行样本 price_column = feature_column.numeric_column('price') bucket_price = feature_column.bucketized_column(price_column, [0, 10, 20, 30, 40]) price_bucket_tensor = feature_column.input_layer(price, [bucket_price]) with tf.Session() as session: print(session.run([price_bucket_tensor])) def test_categorical_column_with_vocabulary_list(): # color_data = {'color': [['R', 'R'], ['G', 'R'], ['B', 'G'], ['A', 'A']]} # 4行样本 color_data = {'color': [['R'], ['R'], ['G'], ['A']]} # 4行样本
Feature_col
最新推荐文章于 2022-03-08 23:19:42 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)