@@ -549,7 +549,7 @@ for i, state_var in enumerate(state_vec1):
549549 ax.plot(λ_vals, F1[:, i], label=r"$\overline{s}_1$", color="b")
550550 ax.plot(λ_vals, F2[:, i], label=r"$\overline{s}_2$", color="r")
551551
552- ax.set_xlabel("$\lambda$")
552+ ax.set_xlabel(r "$\lambda$")
553553 ax.set_ylabel("$F_{s_t}$")
554554 ax.set_title(f"Coefficient on {state_var}")
555555 ax.legend()
@@ -617,8 +617,8 @@ for i, state_var in enumerate(state_vec1):
617617 ax.plot_surface(λ_grid, δ_grid, F1_grid[:, :, i], color="b")
618618 # low adjustment cost, red surface
619619 ax.plot_surface(λ_grid, δ_grid, F2_grid[:, :, i], color="r")
620- ax.set_xlabel("$\lambda$")
621- ax.set_ylabel("$\delta$")
620+ ax.set_xlabel(r "$\lambda$")
621+ ax.set_ylabel(r "$\delta$")
622622 ax.set_zlabel("$F_{s_t}$")
623623 ax.set_title(f"coefficient on {state_var}")
624624 plt.show()
@@ -659,8 +659,8 @@ def run(construct_func, vals_dict, state_vec):
659659 ax.plot(λ_vals, F1[:, i], label=r"$\overline{s}_1$", color="b")
660660 ax.plot(λ_vals, F2[:, i], label=r"$\overline{s}_2$", color="r")
661661
662- ax.set_xlabel("$\lambda$")
663- ax.set_ylabel("$F(\overline{s}_t)$")
662+ ax.set_xlabel(r "$\lambda$")
663+ ax.set_ylabel(r "$F(\overline{s}_t)$")
664664 ax.set_title(f"coefficient on {state_var}")
665665 ax.legend()
666666 plt.show()
@@ -681,10 +681,10 @@ def run(construct_func, vals_dict, state_vec):
681681 # Plot a vertical line at λ=0.5
682682 ax.plot([0.5, 0.5], [min(k_star), max(k_star)], "-.")
683683
684- ax.set_xlabel("$\lambda$")
684+ ax.set_xlabel(r "$\lambda$")
685685 ax.set_ylabel("$k$")
686686 ax.set_title("Optimal k levels and k targets")
687- ax.text(0.5, min(k_star)+(max(k_star)-min(k_star))/20, "$\lambda=0.5$")
687+ ax.text(0.5, min(k_star)+(max(k_star)-min(k_star))/20, r "$\lambda=0.5$")
688688 ax.legend(bbox_to_anchor=(1., 1.))
689689 plt.show()
690690
@@ -714,9 +714,9 @@ def run(construct_func, vals_dict, state_vec):
714714 ax = fig.add_subplot(111, projection='3d')
715715 ax.plot_surface(λ_grid, δ_grid, F1_grid[:, :, i], color="b")
716716 ax.plot_surface(λ_grid, δ_grid, F2_grid[:, :, i], color="r")
717- ax.set_xlabel("$\lambda$")
718- ax.set_ylabel("$\delta$")
719- ax.set_zlabel("$F(\overline{s}_t)$")
717+ ax.set_xlabel(r "$\lambda$")
718+ ax.set_ylabel(r "$\delta$")
719+ ax.set_zlabel(r "$F(\overline{s}_t)$")
720720 ax.set_title(f"coefficient on {state_var}")
721721 plt.show()
722722```
0 commit comments