perpetual 0.9.1__cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 0.9.3__cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

perpetual/booster.py

@@ -65,7 +65,9 @@ class PerpetualBooster:
             objective (str, optional): Learning objective function to be used for optimization. Valid options are:
                 "LogLoss" to use logistic loss (classification),
                 "SquaredLoss" to use squared error (regression),
-                "QuantileLoss" to use quantile error (regression).
+                "QuantileLoss" to use quantile error (regression),
+                "HuberLoss" to use huber error (regression),
+                "AdaptiveHuberLoss" to use adaptive huber error (regression).
                 Defaults to "LogLoss".
             budget (float, optional): a positive number for fitting budget. Increasing this number will more
                 likely result in more boosting rounds and more increased predictive power.
@@ -227,7 +229,7 @@ class PerpetualBooster:
         self.cat_mapping = cat_mapping
         self.feature_names_in_ = features_
 
-        y_, classes_ = convert_input_array(y, self.objective)
+        y_, classes_ = convert_input_array(y, self.objective, is_target=True)
         self.classes_ = np.array(classes_).tolist()
 
         if sample_weight is None:
@@ -488,9 +490,10 @@ class PerpetualBooster:
                 axis=1,
             )
         else:
-            raise NotImplementedError(
+            warnings.warn(
                 f"predict_proba not implemented for regression. n_classes = {len(self.classes_)}"
             )
+            return np.ones((rows, 1))
 
     def predict_log_proba(self, X, parallel: Union[bool, None] = None) -> np.ndarray:
         """Predict class log-probabilities with the fitted booster on new data.
@@ -524,9 +527,8 @@ class PerpetualBooster:
                 parallel=parallel,
             )
         else:
-            raise NotImplementedError(
-                "predict_log_proba not implemented for regression."
-            )
+            warnings.warn("predict_log_proba not implemented for regression.")
+            return np.ones((rows, 1))
 
     def predict_nodes(self, X, parallel: Union[bool, None] = None) -> List:
         """Predict nodes with the fitted booster on new data.

perpetual/utils.py

@@ -32,7 +32,7 @@ def type_series(y):
         return ""
 
 
-def convert_input_array(x, objective) -> np.ndarray:
+def convert_input_array(x, objective, is_target=False) -> np.ndarray:
     classes_ = []
 
     if type(x).__module__.split(".")[0] == "numpy":
@@ -49,7 +49,7 @@ def convert_input_array(x, objective) -> np.ndarray:
     else:
         x_ = x.to_numpy()
 
-    if objective == "LogLoss" and len(x_.shape) == 1:
+    if is_target and objective == "LogLoss" and len(x_.shape) == 1:
         classes_ = np.unique(x_)
         x_index = np.array([np.where(classes_ == i) for i in x_])
         if len(classes_) > 2:

{perpetual-0.9.1.dist-info → perpetual-0.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: perpetual
-Version: 0.9.1
+Version: 0.9.3
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
@@ -45,6 +45,7 @@ Project-URL: Source Code, https://github.com/perpetual-ml/perpetual
 [![Static Badge](https://img.shields.io/badge/join-discord-blue?logo=discord)](https://discord.gg/AyUK7rr6wy)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/perpetual)
 
+
 </div>
 
 # Perpetual

perpetual-0.9.3.dist-info/RECORD

@@ -0,0 +1,11 @@
+perpetual-0.9.3.dist-info/METADATA,sha256=LMtEPPxeAQVXSo8ZFIg-GeyMU8m9lBgChGw5qxSGoBE,10594
+perpetual-0.9.3.dist-info/WHEEL,sha256=LVyF9ZNzxVF0Pv_yNwDscEc-4yPApMsUndxUxnmZZi4,131
+perpetual-0.9.3.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
+perpetual/__init__.py,sha256=zzFLkwRr8No7DPUNUQlBXdsdGfT7KEt4orHozdZhS7c,116
+perpetual/booster.py,sha256=4MX7fahwGk4dKVGqZA-_OINw88F03EEoYLFCabUoms4,49863
+perpetual/data.py,sha256=OCuRbjmaY73fOqm_MnI8mgQxAH70cpSOpYaXauaELvs,605
+perpetual/perpetual.cpython-311-aarch64-linux-gnu.so,sha256=3JxZxiYMapujAhMuuPCgzMMoferUYVqX5j6kg8_UXlM,1709232
+perpetual/serialize.py,sha256=Tg2BbuA1jKQ5-ITuVhwtj6hgBaRAbZ66eHctR7fcVk4,1883
+perpetual/types.py,sha256=vBwsX2XnSC8RqdmP2Mr6hDljgWhgGNS6OsuBZM1QdW0,3231
+perpetual/utils.py,sha256=UCY_w5FNOsgn7NKWiLy2uWzYgDdFRTLgRfIM2iAWf8w,7269
+perpetual-0.9.3.dist-info/RECORD,,

{perpetual-0.9.1.dist-info → perpetual-0.9.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.3)
+Generator: maturin (1.8.6)
 Root-Is-Purelib: false
 Tag: cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64

perpetual-0.9.1.dist-info/RECORD

@@ -1,11 +0,0 @@
-perpetual-0.9.1.dist-info/METADATA,sha256=uI0NWv8HssqmSq5aOaviUpzo177FyyqkAeo-_TuYWkc,10593
-perpetual-0.9.1.dist-info/WHEEL,sha256=Mgc7q3H5chuFJt-w6cZ-qv2aS5dlvenFVMCZtQp77B4,131
-perpetual-0.9.1.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-perpetual/__init__.py,sha256=zzFLkwRr8No7DPUNUQlBXdsdGfT7KEt4orHozdZhS7c,116
-perpetual/types.py,sha256=vBwsX2XnSC8RqdmP2Mr6hDljgWhgGNS6OsuBZM1QdW0,3231
-perpetual/serialize.py,sha256=Tg2BbuA1jKQ5-ITuVhwtj6hgBaRAbZ66eHctR7fcVk4,1883
-perpetual/data.py,sha256=OCuRbjmaY73fOqm_MnI8mgQxAH70cpSOpYaXauaELvs,605
-perpetual/utils.py,sha256=DToduVbdLPT7-41nXIiVbQ_7IXFo4pI0auY-fz6Qz44,7238
-perpetual/booster.py,sha256=XGs2J-SqhCI7T-qZfBgkrE4yQMIy5vy7OBwK8IO3uRs,49686
-perpetual/perpetual.cpython-311-aarch64-linux-gnu.so,sha256=emiN7HlABOEIdxSpzC2EW2rUvp3KhX3yYg23UsJUYeA,1643600
-perpetual-0.9.1.dist-info/RECORD,,