Skip to content

Commit cf38817

Browse files
committed
Refine iteratively
1 parent 160d6d7 commit cf38817

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/penn_chime/models.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,12 +95,13 @@ def __init__(self, p: Parameters):
9595
)
9696

9797
# Make an initial coarse estimate
98-
dts = np.linspace(1, 15, 29)
98+
dts = np.linspace(1, 15, 15)
9999
min_loss = self.get_argmin_doubling_time(p, dts)
100100

101101
# Refine the coarse estimate
102-
dts = np.linspace(dts[min_loss-1], dts[min_loss+1], 29)
103-
min_loss = self.get_argmin_doubling_time(p, dts)
102+
for iteration in range(4):
103+
dts = np.linspace(dts[min_loss-1], dts[min_loss+1], 15)
104+
min_loss = self.get_argmin_doubling_time(p, dts)
104105

105106
p.doubling_time = dts[min_loss]
106107

0 commit comments

Comments
 (0)