move confusion matrix text to tooltip (#127)

iterative · Apr 20, 2023 · 8aab523 · 8aab523
1 parent 33eae51
commit 8aab523
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/dvc_render/vega_templates.py b/src/dvc_render/vega_templates.py
@@ -267,6 +267,7 @@ class ConfusionTemplate(Template):
                         "tooltip": [
                             {"field": Template.anchor("x"), "type": "nominal"},
                             {"field": Template.anchor("y"), "type": "nominal"},
+                            {"field": "xy_count", "type": "quantitative"},
                         ],
                         "opacity": {
                             "condition": {"selection": "label", "value": 1},
@@ -283,7 +284,6 @@ class ConfusionTemplate(Template):
                 {
                     "mark": "text",
                     "encoding": {
-                        "text": {"field": "xy_count", "type": "quantitative"},
                         "color": {
                             "condition": {
                                 "test": "datum.percent_of_max > 0.5",
@@ -377,6 +377,11 @@ class NormalizedConfusionTemplate(Template):
                         "tooltip": [
                             {"field": Template.anchor("x"), "type": "nominal"},
                             {"field": Template.anchor("y"), "type": "nominal"},
+                            {
+                                "field": "percent_of_y",
+                                "type": "quantitative",
+                                "format": ".2f",
+                            },
                         ],
                         "opacity": {
                             "condition": {"selection": "label", "value": 1},
@@ -393,11 +398,6 @@ class NormalizedConfusionTemplate(Template):
                 {
                     "mark": "text",
                     "encoding": {
-                        "text": {
-                            "field": "percent_of_y",
-                            "type": "quantitative",
-                            "format": ".2f",
-                        },
                         "color": {
                             "condition": {
                                 "test": "datum.percent_of_y > 0.5",