vega: templates: make confusion matrix interactive

iterative · May 11, 2022 · 46dc455 · 46dc455
1 parent 28c1495
commit 46dc455
Showing 1 changed file with 56 additions and 0 deletions.
diff --git a/src/dvc_render/vega_templates.py b/src/dvc_render/vega_templates.py
@@ -175,6 +175,34 @@ class ConfusionTemplate(Template):
                         }
                     },
                 },
+                {
+                    "selection": {
+                        "label": {
+                            "type": "single",
+                            "on": "mouseover",
+                            "encodings": ["x", "y"],
+                            "empty": "none",
+                            "clear": "mouseout",
+                        }
+                    },
+                    "mark": "rect",
+                    "encoding": {
+                        "tooltip": [
+                            {"field": "actual", "type": "nominal"},
+                            {"field": "predicted", "type": "nominal"},
+                        ],
+                        "opacity": {
+                            "condition": {"selection": "label", "value": 1},
+                            "value": 0,
+                        },
+                    },
+                },
+                {
+                    "transform": [{"filter": {"selection": "label"}}],
+                    "layer": [
+                        {"mark": {"type": "rect", "color": "lightpink"}},
+                    ],
+                },
                 {
                     "mark": "text",
                     "encoding": {
@@ -257,6 +285,34 @@ class NormalizedConfusionTemplate(Template):
                         }
                     },
                 },
+                {
+                    "selection": {
+                        "label": {
+                            "type": "single",
+                            "on": "mouseover",
+                            "encodings": ["x", "y"],
+                            "empty": "none",
+                            "clear": "mouseout",
+                        }
+                    },
+                    "mark": "rect",
+                    "encoding": {
+                        "tooltip": [
+                            {"field": "actual", "type": "nominal"},
+                            {"field": "predicted", "type": "nominal"},
+                        ],
+                        "opacity": {
+                            "condition": {"selection": "label", "value": 1},
+                            "value": 0,
+                        },
+                    },
+                },
+                {
+                    "transform": [{"filter": {"selection": "label"}}],
+                    "layer": [
+                        {"mark": {"type": "rect", "color": "lightpink"}},
+                    ],
+                },
                 {
                     "mark": "text",
                     "encoding": {