File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -818,6 +818,9 @@ cdef inline double_pair cgrad_hess_exponential(
818818cdef class CyLossFunction:
819819 """Base class for convex loss functions."""
820820
821+ def __reduce__(self):
822+ return (self.__class__, ())
823+
821824 cdef double cy_loss(self, double y_true, double raw_prediction) noexcept nogil:
822825 """Compute the loss for a single sample.
823826
@@ -1013,6 +1016,11 @@ cdef class {{name}}(CyLossFunction):
10131016 self.{{param}} = {{param}}
10141017 {{endif}}
10151018
1019+ {{if param is not None}}
1020+ def __reduce__(self):
1021+ return (self.__class__, (self.{{param}},))
1022+ {{endif}}
1023+
10161024 cdef inline double cy_loss(self, double y_true, double raw_prediction) noexcept nogil:
10171025 return {{closs}}(y_true, raw_prediction{{with_param}})
10181026
You can’t perform that action at this time.
0 commit comments