Skip to content

Commit 55ed62a

Browse files
committed
fix plot doc
1 parent 3be2424 commit 55ed62a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/plot_survival_analysis.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@
6969

7070
# skglm internals: init datafit and penalty
7171
datafit = Cox()
72+
datafit.initialize(X, y)
7273
penalty = L1(alpha)
7374

7475
# init solver
@@ -228,6 +229,7 @@
228229

229230
# ensure using Efron estimate
230231
datafit = Cox(use_efron=True)
232+
datafit.initialize(X, y)
231233

232234
# solve the problem
233235
solver = ProxNewton(fit_intercept=False, max_iter=50)

0 commit comments

Comments
 (0)