Spaces:

samuelinferences
/

transformers-can-do-bayesian-inference

Running

App Files Files Community

Samuel Mueller commited on Nov 3, 2021

Commit

a8d8c56

·

1 Parent(s): 981d877

axis labels

Files changed (1) hide show

app.py +3 -1

app.py CHANGED Viewed

@@ -90,7 +90,9 @@ def infer(table, choice):
     plot_w_conf_interval(plt, test_xs, *mean_and_bounds_for_gp(x,y,test_xs), 'green', 'GP')
     plot_w_conf_interval(plt, test_xs, *mean_and_bounds_for_pnf(x,y,test_xs, choice), 'blue', 'PFN')
-    plt.legend(ncol=2,bbox_to_anchor=[0.5,-.08],loc="upper center")
     plt.tight_layout()

     plot_w_conf_interval(plt, test_xs, *mean_and_bounds_for_gp(x,y,test_xs), 'green', 'GP')
     plot_w_conf_interval(plt, test_xs, *mean_and_bounds_for_pnf(x,y,test_xs, choice), 'blue', 'PFN')
+    plt.legend(ncol=2,bbox_to_anchor=[0.5,-.14],loc="upper center")
+    plt.xlabel('x')
+    plt.ylabel('y')
     plt.tight_layout()