From 715e065059365922f9d3b3d6d543f92789de5bd5 Mon Sep 17 00:00:00 2001 From: Renmin Gu Date: Wed, 1 Apr 2020 19:58:02 +0800 Subject: [PATCH] enable native Keras + TFMA --- samples/core/iris/iris.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/samples/core/iris/iris.py b/samples/core/iris/iris.py index 0c00d9a49b0..fe099dd7dce 100644 --- a/samples/core/iris/iris.py +++ b/samples/core/iris/iris.py @@ -137,7 +137,10 @@ def _create_pipeline( name='baseline', label_key='variety', is_baseline=True ) ], - slicing_specs=[tfma.SlicingSpec()], + slicing_specs=[ + tfma.SlicingSpec(), + # Data can be sliced along a feature column. Required by TFMA visualization. + tfma.SlicingSpec(feature_keys=['sepal_length'])], metrics_specs=[ tfma.MetricsSpec( metrics=[