deskit 1.3__tar.gz → 1.3.1__tar.gz

{deskit-1.3/src/deskit.egg-info → deskit-1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deskit
-Version: 1.3
+Version: 1.3.1
 Summary: A Python library for Dynamic Ensemble Selection
 Author: Tikhon Vodyanov
 License-Expression: MIT
@@ -251,6 +251,8 @@ router = DEWSU(task="regression", metric=pinball, mode="min", k=20)
 
 Built-in metric strings: `accuracy`, `mae`, `mse`, `rmse`, `log_loss`, `prob_correct`.
 
+Alternatively, pass `metric=None` and populate the dictionary passed to fit() with your own per-sample computed scores instead of predictions.
+
 ---
 
 ## Data types

{deskit-1.3 → deskit-1.3.1}/README.md

@@ -220,6 +220,8 @@ router = DEWSU(task="regression", metric=pinball, mode="min", k=20)
 
 Built-in metric strings: `accuracy`, `mae`, `mse`, `rmse`, `log_loss`, `prob_correct`.
 
+Alternatively, pass `metric=None` and populate the dictionary passed to fit() with your own per-sample computed scores instead of predictions.
+
 ---
 
 ## Data types

{deskit-1.3 → deskit-1.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "deskit"
-version = "1.3"
+version = "1.3.1"
 description = "A Python library for Dynamic Ensemble Selection"
 readme = "README.md"
 license = "MIT"

{deskit-1.3 → deskit-1.3.1}/src/deskit/base/knnbase.py

@@ -49,12 +49,21 @@ class KNNBase(PredictBase, BaseRouter):
         This method expects pre-validated numpy arrays.
         """
         self.models = list(preds_dict.keys())
-        n_val       = len(y)
-        n_models    = len(self.models)
+        n_val = len(y)
+        n_models = len(self.models)
         self.matrix = np.zeros((n_val, n_models))
 
         for j, name in enumerate(self.models):
-            scores = self._compute_scores(y, preds_dict[name])
+            if self.metric is not None:
+                scores = self._compute_scores(y, preds_dict[name])
+            else:
+                scores = np.asarray(preds_dict[name])
+                # Validate shape
+                if scores.ndim != 1 or len(scores) != n_val:
+                    raise ValueError(
+                        f"Raw scores for model '{name}' must be a 1D array of length {n_val}; "
+                        f"got shape {scores.shape}."
+                    )
             self.matrix[:, j] = scores if self.mode == 'max' else -scores
 
         if self.task == 'classification':

{deskit-1.3 → deskit-1.3.1/src/deskit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deskit
-Version: 1.3
+Version: 1.3.1
 Summary: A Python library for Dynamic Ensemble Selection
 Author: Tikhon Vodyanov
 License-Expression: MIT
@@ -251,6 +251,8 @@ router = DEWSU(task="regression", metric=pinball, mode="min", k=20)
 
 Built-in metric strings: `accuracy`, `mae`, `mse`, `rmse`, `log_loss`, `prob_correct`.
 
+Alternatively, pass `metric=None` and populate the dictionary passed to fit() with your own per-sample computed scores instead of predictions.
+
 ---
 
 ## Data types