mlquantify 0.1.15__tar.gz → 0.1.17__tar.gz

{mlquantify-0.1.15/mlquantify.egg-info → mlquantify-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlquantify
-Version: 0.1.15
+Version: 0.1.17
 Summary: Quantification Library
 Home-page: https://github.com/luizfernandolj/QuantifyML/tree/master
 Maintainer: Luiz Fernando Luth Junior

mlquantify-0.1.17/VERSION.txt

@@ -0,0 +1 @@
+0.1.17

{mlquantify-0.1.15 → mlquantify-0.1.17}/mlquantify/adjust_counting/_adjustment.py

@@ -105,7 +105,8 @@ class ThresholdAdjustment(SoftLearnerQMixin, BaseAdjustCount):
         thresholds, tprs, fprs = evaluate_thresholds(train_y_values, positive_scores)
         threshold, tpr, fpr = self.get_best_threshold(thresholds, tprs, fprs)
 
-        cc_predictions = CC(threshold=threshold).aggregate(predictions, train_y_values)[1]
+        cc_predictions = CC(threshold=threshold).aggregate(predictions, train_y_values)
+        cc_predictions = list(cc_predictions.values())[1]
 
         if tpr - fpr == 0:
             prevalence = cc_predictions
@@ -204,13 +205,13 @@ class MatrixAdjustment(BaseAdjustCount):
         self.CM = np.zeros((n_class, n_class))
 
         if self.solver == 'optim':
-            priors = np.array(list(CC().aggregate(train_y_pred).values()))
+            priors = np.array(list(CC().aggregate(train_y_pred, train_y_values).values()))
             self.CM = self._compute_confusion_matrix(train_y_pred, train_y_values, priors)
-            prevs_estim = self._get_estimations(predictions > priors)
+            prevs_estim = self._get_estimations(predictions > priors, train_y_values)
             prevalence = self._solve_optimization(prevs_estim, priors)
         else:
             self.CM = self._compute_confusion_matrix(train_y_pred, train_y_values)
-            prevs_estim = self._get_estimations(predictions)
+            prevs_estim = self._get_estimations(predictions, train_y_values)
             prevalence = self._solve_linear(prevs_estim)
         
         return prevalence
@@ -261,11 +262,11 @@ class MatrixAdjustment(BaseAdjustCount):
         result = minimize(objective, init, constraints=constraints, bounds=bounds)
         return result.x if result.success else priors
 
-    def _get_estimations(self, predictions):
+    def _get_estimations(self, predictions, train_y_values):
         """Return prevalence estimates using CC (crisp) or PCC (probabilistic)."""
         if uses_soft_predictions(self):
             return np.array(list(PCC().aggregate(predictions).values()))
-        return np.array(list(CC().aggregate(predictions).values()))
+        return np.array(list(CC().aggregate(predictions, train_y_values).values()))
 
     @abstractmethod
     def _compute_confusion_matrix(self, predictions, *args):
@@ -609,7 +610,7 @@ class MS(ThresholdAdjustment):
         prevs = []
         for thr, tpr, fpr in zip(thresholds, tprs, fprs):
             cc_predictions = CC(threshold=thr).aggregate(predictions, train_y_values)
-            cc_predictions = cc_predictions[1]
+            cc_predictions = list(cc_predictions.values())[1]
             
             if tpr - fpr == 0:
                 prevalence = cc_predictions

{mlquantify-0.1.15 → mlquantify-0.1.17}/mlquantify/adjust_counting/_counting.py

@@ -76,14 +76,15 @@ class CC(CrispLearnerQMixin, BaseCount):
         self.threshold = threshold
 
     def aggregate(self, predictions, train_y_values=None):
-        predictions = validate_predictions(self, predictions, self.threshold)
+        predictions = validate_predictions(self, predictions, self.threshold, train_y_values)
         
         if train_y_values is None:
             train_y_values = np.unique(predictions)
+            
         self.classes_ = check_classes_attribute(self, np.unique(train_y_values))
         class_counts = np.array([np.count_nonzero(predictions == _class) for _class in self.classes_])
         prevalences = class_counts / len(predictions)
-        
+
         prevalences = validate_prevalences(self, prevalences, self.classes_)
         return prevalences
 

{mlquantify-0.1.15 → mlquantify-0.1.17}/mlquantify/utils/_validation.py

@@ -96,23 +96,49 @@ def validate_y(quantifier: Any, y: np.ndarray) -> None:
 
 def _get_valid_crisp_predictions(predictions, threshold=0.5):
     predictions = np.asarray(predictions)
-
     dimensions = predictions.ndim
 
+    if train_y_values is not None:
+        classes = np.unique(train_y_values)
+    else:
+        classes = None
+
     if dimensions > 2:
-        predictions = np.argmax(predictions, axis=1)
+        # Assuming the last dimension contains class probabilities
+        crisp_indices = np.argmax(predictions, axis=-1)
+        if classes is not None:
+            predictions = classes[crisp_indices]
+        else:
+            predictions = crisp_indices
     elif dimensions == 2:
-        predictions = (predictions[:, 1] >= threshold).astype(int)
+        # Binary or multi-class probabilities (N, C)
+        if classes is not None and len(classes) == 2:
+            # Binary case with explicit classes
+            predictions = np.where(predictions[:, 1] >= threshold, classes[1], classes[0])
+        elif classes is not None and len(classes) > 2:
+            # Multi-class case with explicit classes
+            crisp_indices = np.argmax(predictions, axis=1)
+            predictions = classes[crisp_indices]
+        else:
+            # Default binary (0 or 1) or multi-class (0 to C-1)
+            if predictions.shape[1] == 2:
+                predictions = (predictions[:, 1] >= threshold).astype(int)
+            else:
+                predictions = np.argmax(predictions, axis=1)
     elif dimensions == 1:
+        # 1D probabilities (e.g., probability of positive class)
         if np.issubdtype(predictions.dtype, np.floating):
-            predictions = (predictions >= threshold).astype(int)
+            if classes is not None and len(classes) == 2:
+                predictions = np.where(predictions >= threshold, classes[1], classes[0])
+            else:
+                predictions = (predictions >= threshold).astype(int)
     else:
         raise ValueError(f"Predictions array has an invalid number of dimensions. Expected 1 or more dimensions, got {predictions.ndim}.")
 
     return predictions
 
 
-def validate_predictions(quantifier: Any, predictions: np.ndarray, threshold: float = 0.5) -> np.ndarray:
+def validate_predictions(quantifier: Any, predictions: np.ndarray, threshold: float = 0.5, train_y_values=None) -> np.ndarray:
     """
     Validate predictions using the quantifier's declared output tags.
     Raises InputValidationError if inconsistent with tags.
@@ -132,7 +158,7 @@ def validate_predictions(quantifier: Any, predictions: np.ndarray, threshold: fl
             f"Soft predictions for {quantifier.__class__.__name__} must be float, got dtype {predictions.dtype}."
         )
     elif estimator_type == "crisp" and np.issubdtype(predictions.dtype, np.floating):
-        predictions = _get_valid_crisp_predictions(predictions, threshold) 
+        predictions = _get_valid_crisp_predictions(predictions, train_y_values, threshold) 
     return predictions   
     
     

{mlquantify-0.1.15 → mlquantify-0.1.17/mlquantify.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlquantify
-Version: 0.1.15
+Version: 0.1.17
 Summary: Quantification Library
 Home-page: https://github.com/luizfernandolj/QuantifyML/tree/master
 Maintainer: Luiz Fernando Luth Junior

mlquantify-0.1.15/VERSION.txt

@@ -1 +0,0 @@
-0.1.15