MNT add __reduce__ to loss objects (#30356)

adrinjalali · jeremiedbb · commit 26384be3485c · 2024-12-06T22:23:50.000+01:00
diff --git a/sklearn/_loss/_loss.pyx.tp b/sklearn/_loss/_loss.pyx.tp
@@ -818,6 +818,9 @@ cdef inline double_pair cgrad_hess_exponential(
 cdef class CyLossFunction:
     """Base class for convex loss functions."""
 
+    def __reduce__(self):
+        return (self.__class__, ())
+
     cdef double cy_loss(self, double y_true, double raw_prediction) noexcept nogil:
         """Compute the loss for a single sample.
 
@@ -1013,6 +1016,11 @@ cdef class {{name}}(CyLossFunction):
         self.{{param}} = {{param}}
     {{endif}}
 
+    {{if param is not None}}
+    def __reduce__(self):
+        return (self.__class__, (self.{{param}},))
+    {{endif}}
+
     cdef inline double cy_loss(self, double y_true, double raw_prediction) noexcept nogil:
         return {{closs}}(y_true, raw_prediction{{with_param}})
 
