Spaces:

yp-edu
/

viz-gpt2-stockfish-debug

Sleeping

Xmaster6y commited on Feb 29

Commit

1da6233

•

1 Parent(s): 8694e50

smaller colorbar

Files changed (1) hide show

src/visualisation.py CHANGED Viewed

@@ -41,7 +41,7 @@ def render_heatmap(
         color = COLOR_MAP(norm(heatmap[square_index]))
         color = (*color[:3], ALPHA)
         color_dict[square_index] = matplotlib.colors.to_hex(color, keep_alpha=True)
-    fig = plt.figure(figsize=(6, 2))
     ax = plt.gca()
     ax.axis("off")
     fig.colorbar(

         color = COLOR_MAP(norm(heatmap[square_index]))
         color = (*color[:3], ALPHA)
         color_dict[square_index] = matplotlib.colors.to_hex(color, keep_alpha=True)
+    fig = plt.figure(figsize=(6, 1))
     ax = plt.gca()
     ax.axis("off")
     fig.colorbar(