test_feature_transforms.py 文件源码

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