legend interactive

app.py

@@ -62,7 +62,7 @@ def down_samp(embedding):
 
 
 def data_comparison(df):
-    selection = alt.selection_multi(fields=['cluster:N','label:O'])
+    selection = alt.selection_multi(fields=['cluster','label'])
     color = alt.condition(alt.datum.slice == 'high-loss', alt.Color('cluster:N', scale = alt.Scale(domain=df.cluster.unique().tolist())), alt.value("lightgray"))
     opacity = alt.condition(selection, alt.value(0.7), alt.value(0.25))
 
@@ -71,8 +71,8 @@ def data_comparison(df):
         x=alt.X('x:Q', axis=None),
         y=alt.Y('y:Q', axis=None),
         color=color,
-        shape=alt.Shape('label:O', scale=alt.Scale(range=['circle', 'diamond'])),
-        tooltip=['cluster:N','slice:N','content:N','label:O','pred:O'],
+        shape=alt.Shape('label:N', scale=alt.Scale(range=['circle', 'diamond'])),
+        tooltip=['cluster:N','slice:N','content:N','label:N','pred:O'],
         opacity=opacity
     ).properties(
         width=1000,
@@ -80,9 +80,9 @@ def data_comparison(df):
     ).interactive()
 
     legend = alt.Chart(df).mark_point(size=100, filled=True).encode(
-        x=alt.X("label:O"),
+        x=alt.X("label"),
         y=alt.Y('cluster:N', axis=alt.Axis(orient='right'), title=""),
-        shape=alt.Shape('label:O', scale=alt.Scale(
+        shape=alt.Shape('label:N', scale=alt.Scale(
         range=['circle', 'diamond']), legend=None),
         color=color,
     ).add_selection(