555e7d8f67b8426556aaa40fb0a248de3a40171e,solutionbox/code_free_ml/mltoolbox/code_free_ml/data/analyze_data.py,,make_transform_graph,#Any#Any#Any#,651

Before Change

  preprocessing_fn = make_preprocessing_fn(output_dir, features)

  // copy from /tft/beam/impl
  inputs, outputs = impl_helper.run_preprocessing_fn(
      preprocessing_fn=preprocessing_fn,
      schema=tft_input_schema)
  output_metadata = dataset_metadata.DatasetMetadata(

After Change

    with tft_impl.Context(temp_dir=temp_dir):

      // Not going to transform, so no data is needed.
      train_data = p | beam.Create([])

      transform_fn = (
        (train_data, tft_input_metadata)
        | "BuildTransformFn"  // noqa
        >> tft_impl.AnalyzeDataset(preprocessing_fn))  // noqa

      // Writes transformed_metadata and transfrom_fn folders
      _ = (transform_fn | "WriteTransformFn" >> tft_beam_io.WriteTransformFn(output_dir))  // noqa

Instances