Skip to content
This repository was archived by the owner on Dec 6, 2023. It is now read-only.

Commit f06f871

Browse files
committed
fix pickling losses in primal_cd
1 parent 9cde4bb commit f06f871

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

lightning/impl/primal_cd_fast.pyx

+6
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,12 @@ cdef class LossFunction:
3232
cdef double beta
3333
cdef int verbose
3434

35+
def __getstate__(self):
36+
return self.max_steps, self.sigma, self.beta, self.verbose
37+
38+
def __setstate__(self, state):
39+
self.max_steps, self.sigma, self.beta, self.verbose = state
40+
3541
# L2 regularization
3642

3743
cdef void solve_l2(self,

lightning/impl/tests/test_primal_cd.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,8 @@ def test_debiasing_l1l2():
178178
multiclass=False,
179179
debiasing=True,
180180
warm_debiasing=warm_debiasing,
181-
max_iter=20, C=0.01, random_state=0)
181+
max_iter=20, C=0.01, random_state=0,
182+
verbose=True)
182183
clf.fit(mult_csc, mult_target)
183184
assert_greater(clf.score(mult_csc, mult_target), 0.75)
184185
assert_equal(clf.n_nonzero(percentage=True), 0.08)

0 commit comments

Comments
 (0)