enable native Keras + TFMA (kubeflow#3409)

Co-authored-by: Renmin Gu <renming@google.com>
Jeffwan · Dec 9, 2020 · d2428d4 · d2428d4
1 parent 2b9b220
commit d2428d4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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=[