@@ -511,6 +511,8 @@ m_seq_2_regime1 = np.concatenate((m_seq_2_path1[:T1],
511
511
p_seq_2_regime1 = np.concatenate((p_seq_2_path1[:T1],
512
512
p_seq_2_cont1))
513
513
514
+ π_seq_2[T1-1] = p_seq_2_regime1[T1] - p_seq_2_regime1[T1-1]
515
+
514
516
# regime 2 - reset m_T1
515
517
m_T1 = (m_seq_2_path1[T1-1] + μ0) + cm2.α*(μ0 - μ_star)
516
518
@@ -540,8 +542,8 @@ plot_configs = [
540
542
{'data': [(T_seq, m_seq_2_regime1, 'Smooth $m_{T_1}$'),
541
543
(T_seq, m_seq_2_regime2, 'Jumpy $m_{T_1}$')],
542
544
'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}$')],
545
547
'ylabel': r'$p$'}
546
548
]
547
549
@@ -592,7 +594,7 @@ fig, ax = plt.subplots(5, figsize=(5, 12), dpi=200)
592
594
plot_configs = [
593
595
{'data': [(T_seq[:-1], μ_seq_2)], 'ylabel': r'$\mu$'},
594
596
{'data': [(T_seq, π_seq_2, 'Unforeseen'),
595
- (T_seq, π_seq_1, 'Foreseen')], 'ylabel': r'$p $'},
597
+ (T_seq, π_seq_1, 'Foreseen')], 'ylabel': r'$\pi $'},
596
598
{'data': [(T_seq, m_seq_2_regime1 - p_seq_2_regime1, 'Unforeseen'),
597
599
(T_seq, m_seq_1 - p_seq_1, 'Foreseen')], 'ylabel': r'$m - p$'},
598
600
{'data': [(T_seq, m_seq_2_regime1, 'Unforeseen (Smooth $m_{T_1}$)'),
0 commit comments