@@ -511,6 +511,8 @@ m_seq_2_regime1 = np.concatenate((m_seq_2_path1[:T1],
511511p_seq_2_regime1 = np.concatenate((p_seq_2_path1[:T1],
512512 p_seq_2_cont1))
513513
514+ π_seq_2[T1-1] = p_seq_2_regime1[T1] - p_seq_2_regime1[T1-1]
515+
514516# regime 2 - reset m_T1
515517m_T1 = (m_seq_2_path1[T1-1] + μ0) + cm2.α*(μ0 - μ_star)
516518
@@ -540,8 +542,8 @@ plot_configs = [
540542 {'data': [(T_seq, m_seq_2_regime1, 'Smooth $m_{T_1}$'),
541543 (T_seq, m_seq_2_regime2, 'Jumpy $m_{T_1}$')],
542544 'ylabel': r'$m$'},
543- {'data': [(T_seq, p_seq_2_regime1, 'Smooth $p_ {T_1}$'),
544- (T_seq, p_seq_2_regime2, 'Jumpy $p_ {T_1}$')],
545+ {'data': [(T_seq, p_seq_2_regime1, 'Jumpy $m_ {T_1}$'),
546+ (T_seq, p_seq_2_regime2, 'Smooth $m_ {T_1}$')],
545547 'ylabel': r'$p$'}
546548]
547549
@@ -592,7 +594,7 @@ fig, ax = plt.subplots(5, figsize=(5, 12), dpi=200)
592594plot_configs = [
593595 {'data': [(T_seq[:-1], μ_seq_2)], 'ylabel': r'$\mu$'},
594596 {'data': [(T_seq, π_seq_2, 'Unforeseen'),
595- (T_seq, π_seq_1, 'Foreseen')], 'ylabel': r'$p $'},
597+ (T_seq, π_seq_1, 'Foreseen')], 'ylabel': r'$\pi $'},
596598 {'data': [(T_seq, m_seq_2_regime1 - p_seq_2_regime1, 'Unforeseen'),
597599 (T_seq, m_seq_1 - p_seq_1, 'Foreseen')], 'ylabel': r'$m - p$'},
598600 {'data': [(T_seq, m_seq_2_regime1, 'Unforeseen (Smooth $m_{T_1}$)'),
0 commit comments