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(