google / nitroml

NitroML is a modular, portable, and scalable model-quality benchmarking framework for Machine Learning and Automated Machine Learning (AutoML) pipelines.
Apache License 2.0
41 stars 6 forks source link

Create custom Transform component that can take Schema in preprocessing_fn #12

Closed cweill closed 4 years ago

cweill commented 4 years ago

Similar to our internal version of Transform. The current TFT preprocessing_fn is too restricted.

Related: https://github.com/tensorflow/tfx/issues/687

cweill commented 4 years ago

Fixed with https://github.com/googleinterns/nitroml/commit/70e1e16b8018fbd823e0bf9f91f301d6e254e8f6