vega: templates: make confusion matrix interactive

iterative · May 10, 2022 · 793fe3a · 793fe3a
1 parent 28593d2
commit 793fe3a
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/src/dvc_render/vega_templates.py b/src/dvc_render/vega_templates.py
@@ -175,6 +175,30 @@ class ConfusionTemplate(Template):
                         }
                     },
                 },
+                {
+                    "selection": {
+                        "label": {
+                            "type": "single",
+                            "on": "mouseover",
+                            "encodings": ["x", "y"],
+                            "empty": "none",
+                            "clear": "mouseout",
+                        }
+                    },
+                    "mark": "rect",
+                    "encoding": {
+                        "opacity": {
+                            "condition": {"selection": "label", "value": 1},
+                            "value": 0,
+                        }
+                    },
+                },
+                {
+                    "transform": [{"filter": {"selection": "label"}}],
+                    "layer": [
+                        {"mark": {"type": "rect", "color": "lightpink"}},
+                    ],
+                },
                 {
                     "mark": "text",
                     "encoding": {
@@ -257,6 +281,30 @@ class NormalizedConfusionTemplate(Template):
                         }
                     },
                 },
+                {
+                    "selection": {
+                        "label": {
+                            "type": "single",
+                            "on": "mouseover",
+                            "encodings": ["x", "y"],
+                            "empty": "none",
+                            "clear": "mouseout",
+                        }
+                    },
+                    "mark": "rect",
+                    "encoding": {
+                        "opacity": {
+                            "condition": {"selection": "label", "value": 1},
+                            "value": 0,
+                        }
+                    },
+                },
+                {
+                    "transform": [{"filter": {"selection": "label"}}],
+                    "layer": [
+                        {"mark": {"type": "rect", "color": "lightpink"}},
+                    ],
+                },
                 {
                     "mark": "text",
                     "encoding": {