msm commited on
Commit
e45f29b
·
1 Parent(s): 0256982

legend fixes

Browse files
Files changed (1) hide show
  1. app.py +25 -7
app.py CHANGED
@@ -259,7 +259,10 @@ def create_heatmap(data_pd, metric, color_scheme):
259
  .mark_rect()
260
  .encode(
261
  x=alt.X(
262
- "opponent:N", title="Opponent", sort=unique_models
 
 
 
263
  ), # Ensure consistent sorting
264
  y=alt.Y(
265
  "model:N", title="Model", sort=unique_models
@@ -445,7 +448,12 @@ try:
445
  alt.Chart(df_models.to_pandas())
446
  .mark_bar()
447
  .encode(
448
- x=alt.X("model:N", title="Model", sort=unique_models),
 
 
 
 
 
449
  y=alt.Y("count():Q", title="Count"),
450
  color=alt.Color(
451
  "outcome:N",
@@ -495,7 +503,12 @@ try:
495
  alt.Chart(df_rates_long.to_pandas())
496
  .mark_bar()
497
  .encode(
498
- x=alt.X("model:N", title="Model", sort=unique_models),
 
 
 
 
 
499
  y=alt.Y(
500
  "rate_value:Q",
501
  title="Rate",
@@ -630,7 +643,7 @@ try:
630
  x=alt.X(
631
  "model:N",
632
  title="Model",
633
- axis=alt.Axis(labels=True, labelAngle=90, labelLimit=0),
634
  ), # Show labels, rotated
635
  # --- Choose one Y encoding ---
636
  # For absolute counts (like example):
@@ -737,7 +750,12 @@ try:
737
  .mark_rect()
738
  .encode(
739
  x=alt.X("turn:O", title="Turn"), # Treat turn as ordinal
740
- y=alt.Y("model:N", title="Model", sort=unique_models),
 
 
 
 
 
741
  color=alt.Color(
742
  "win_rate:Q",
743
  title="Win Rate",
@@ -822,7 +840,7 @@ try:
822
  "model:N",
823
  title="Model",
824
  sort=unique_models,
825
- axis=alt.Axis(labelLimit=0, labelAngle=90),
826
  ), # Keep axis formatting
827
  y=alt.Y(
828
  "response_length:Q",
@@ -872,7 +890,7 @@ try:
872
  "model:N",
873
  title="Model",
874
  sort=unique_models,
875
- axis=alt.Axis(labelLimit=0, labelAngle=90),
876
  ), # Keep axis formatting
877
  y=alt.Y(
878
  "num_citations:Q",
 
259
  .mark_rect()
260
  .encode(
261
  x=alt.X(
262
+ "opponent:N",
263
+ title="Opponent",
264
+ sort=unique_models,
265
+ axis=alt.Axis(labelLimit=0, labelAngle=90),
266
  ), # Ensure consistent sorting
267
  y=alt.Y(
268
  "model:N", title="Model", sort=unique_models
 
448
  alt.Chart(df_models.to_pandas())
449
  .mark_bar()
450
  .encode(
451
+ x=alt.X(
452
+ "model:N",
453
+ title="Model",
454
+ sort=unique_models,
455
+ axis=alt.Axis(labelLimit=0),
456
+ ),
457
  y=alt.Y("count():Q", title="Count"),
458
  color=alt.Color(
459
  "outcome:N",
 
503
  alt.Chart(df_rates_long.to_pandas())
504
  .mark_bar()
505
  .encode(
506
+ x=alt.X(
507
+ "model:N",
508
+ title="Model",
509
+ sort=unique_models,
510
+ axis=alt.Axis(labelLimit=0),
511
+ ),
512
  y=alt.Y(
513
  "rate_value:Q",
514
  title="Rate",
 
643
  x=alt.X(
644
  "model:N",
645
  title="Model",
646
+ axis=alt.Axis(labels=True, labelLimit=0),
647
  ), # Show labels, rotated
648
  # --- Choose one Y encoding ---
649
  # For absolute counts (like example):
 
750
  .mark_rect()
751
  .encode(
752
  x=alt.X("turn:O", title="Turn"), # Treat turn as ordinal
753
+ y=alt.Y(
754
+ "model:N",
755
+ title="Model",
756
+ sort=unique_models,
757
+ axis=alt.Axis(labelLimit=0),
758
+ ),
759
  color=alt.Color(
760
  "win_rate:Q",
761
  title="Win Rate",
 
840
  "model:N",
841
  title="Model",
842
  sort=unique_models,
843
+ axis=alt.Axis(labelLimit=0),
844
  ), # Keep axis formatting
845
  y=alt.Y(
846
  "response_length:Q",
 
890
  "model:N",
891
  title="Model",
892
  sort=unique_models,
893
+ axis=alt.Axis(labelLimit=0),
894
  ), # Keep axis formatting
895
  y=alt.Y(
896
  "num_citations:Q",