toggle confusion matrix text

iterative · Oct 16, 2023 · fc0ee0a · fc0ee0a
1 parent e04c5de
commit fc0ee0a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/dvc_render/vega_templates.py b/src/dvc_render/vega_templates.py
@@ -205,6 +205,13 @@ class ConfusionTemplate(Template):
         "data": {"values": Template.anchor("data")},
         "title": Template.anchor("title"),
         "facet": {"field": "rev", "type": "nominal"},
+        "params": [
+            {
+                "name": "toggleText",
+                "bind": {"input": "checkbox"},
+                "title": "Toggle cell value text",
+            },
+        ],
         "spec": {
             "transform": [
                 {
@@ -294,6 +301,13 @@ class ConfusionTemplate(Template):
                 {
                     "mark": "text",
                     "encoding": {
+                        "text": {
+                            "condition": {
+                                "param": "toggleText",
+                                "field": "xy_count",
+                                "type": "quantitative",
+                            },
+                        },
                         "color": {
                             "condition": {
                                 "test": "datum.percent_of_max > 0.5",