From 7efbe20de5e28648d252bb9d1003c75fac19765e Mon Sep 17 00:00:00 2001 From: Daisuke Oyama Date: Fri, 30 Jun 2017 10:57:42 +0900 Subject: [PATCH] Fix career_vf_plot.py --- career/career_vf_plot.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/career/career_vf_plot.py b/career/career_vf_plot.py index 7fcca9d..8eaba64 100644 --- a/career/career_vf_plot.py +++ b/career/career_vf_plot.py @@ -13,10 +13,16 @@ import quantecon as qe from career import CareerWorkerProblem +# === set matplotlib parameters === # +plt.rcParams['axes.xmargin'] = 0 +plt.rcParams['axes.ymargin'] = 0 +plt.rcParams['patch.force_edgecolor'] = True + # === solve for the value function === # wp = CareerWorkerProblem() v_init = np.ones((wp.N, wp.N))*100 -v = qe.compute_fixed_point(wp.bellman_operator, v_init) +v = qe.compute_fixed_point(wp.bellman_operator, v_init, + max_iter=200, print_skip=25) # === plot value function === # fig = plt.figure(figsize=(8, 6)) @@ -32,4 +38,5 @@ ax.set_zlim(150, 200) ax.set_xlabel('theta', fontsize=14) ax.set_ylabel('epsilon', fontsize=14) +ax.view_init(ax.elev, 225) plt.show()