def _run_graph(self, analysis_path, features, schema, stats, predict_data):
"""Runs the preprocessing graph.
Args:
analysis_path: path to folder containing analysis output. Should contain
the stats file.
features: features dict
schema: schema list
stats: stats dict
predict_data: list of csv strings
"""
stats = {'column_stats': {}}
with tf.Graph().as_default():
with tf.Session().as_default() as session:
outputs, labels, inputs = feature_transforms.build_csv_serving_tensors_for_transform_step(
analysis_path, features, schema, stats, keep_target=False)
feed_inputs = {inputs['csv_example']: predict_data}
session.run(tf.tables_initializer())
result = session.run(outputs, feed_dict=feed_inputs)
return result
test_feature_transforms.py 文件源码
python
阅读 24
收藏 0
点赞 0
评论 0
评论列表
文章目录