mlquantify 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

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
@@ -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/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/utils/_validation.py

@@ -94,25 +94,51 @@ def validate_y(quantifier: Any, y: np.ndarray) -> None:
             f"Predictions must be 1D or 2D array, got array with ndim={y.ndim} and shape={y.shape}."
         )
 
-def _get_valid_crisp_predictions(predictions, threshold=0.5):
+def _get_valid_crisp_predictions(predictions, train_y_values=None, 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.16.dist-info → mlquantify-0.1.18.dist-info}/METADATA

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

{mlquantify-0.1.16.dist-info → mlquantify-0.1.18.dist-info}/RECORD

@@ -5,9 +5,9 @@ mlquantify/calibration.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
 mlquantify/confidence.py,sha256=QkEWr6s-Su3Nbinia_TRQbBeTM6ymDPe7Bv204XBKKA,10799
 mlquantify/multiclass.py,sha256=wFbbXKqGsFVSsI9zC0EHGYyyx1JRxFpzMi_q8l80TUM,11770
 mlquantify/adjust_counting/__init__.py,sha256=AWio99zeaUULQq9vKggkFhnq-tqgXxasQt167NdcNVY,307
-mlquantify/adjust_counting/_adjustment.py,sha256=412kFnx3-noaA9u9AatuGIvJbKze-PLPFfBFMBVmQVA,23635
+mlquantify/adjust_counting/_adjustment.py,sha256=YGZiaGBdlWaw7vxGmSfgtWRSqU7Ppc9Lh1eoxmAbAX8,23705
 mlquantify/adjust_counting/_base.py,sha256=MjBsNG7wE0Z_KToXX8WbthhVvz-yc0-d2zIqPo1CB9g,9429
-mlquantify/adjust_counting/_counting.py,sha256=6PKea54xvsga8spNEbsngKNQPyGUXzOkCRyXQR8rTdo,5699
+mlquantify/adjust_counting/_counting.py,sha256=pAWfKK2mHiWAtyuhlld51CnMJFFfMo_X_3IEt5LfCdw,5720
 mlquantify/adjust_counting/_utils.py,sha256=DEPNzvcr0KszCnfUJaRzBilwWzuNVMSdy5eV7aQ_JPE,2907
 mlquantify/likelihood/__init__.py,sha256=3dC5uregNmquUKz0r0-3aPspfjZjKGn3TRBoZPO1uFs,53
 mlquantify/likelihood/_base.py,sha256=seu_Vb58QttcGbFjHKAplMYGZcVbIHqkyTXEK2cax9A,5830
@@ -45,9 +45,9 @@ mlquantify/utils/_parallel.py,sha256=XotpX9nsj6nW-tNCmZ-ahTcRztgnn9oQKP2cl1rLdYM
 mlquantify/utils/_random.py,sha256=7F3nyy7Pa_kN8xP8P1L6MOM4WFu4BirE7bOfGTZ1Spk,1275
 mlquantify/utils/_sampling.py,sha256=QQxE2WKLdiCFUfPF6fKgzyrsOUIWYf74w_w8fbYVc2c,8409
 mlquantify/utils/_tags.py,sha256=Rz78TLpxgVxBKS0mKTlC9Qo_kn6HaEwVKNXh8pxFT7M,1095
-mlquantify/utils/_validation.py,sha256=TGGnfv7F5rnQmVeSqGMuS9AP76O974b1TPishKCCWls,16800
+mlquantify/utils/_validation.py,sha256=zn4OHfa704YBaPKskhiThUG7wS5fvDoHBpcEgb1i8qM,18078
 mlquantify/utils/prevalence.py,sha256=FXLCJViQb2yDbyTXeGZt8WsPPnSZINhorQYZTKXOn14,1772
-mlquantify-0.1.16.dist-info/METADATA,sha256=RZWNq8k48KnN4PYd-6Iw5mv-qS6bAVL2tc6zg9cMqN0,4701
-mlquantify-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mlquantify-0.1.16.dist-info/top_level.txt,sha256=tGEkYkbbFElwULvqENjam3u1uXtyC1J9dRmibsq8_n0,11
-mlquantify-0.1.16.dist-info/RECORD,,
+mlquantify-0.1.18.dist-info/METADATA,sha256=XrQ188Icw5RZEAN8tvHRHTsRm1IKB1iwR_tm6G7uB0w,4701
+mlquantify-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mlquantify-0.1.18.dist-info/top_level.txt,sha256=tGEkYkbbFElwULvqENjam3u1uXtyC1J9dRmibsq8_n0,11
+mlquantify-0.1.18.dist-info/RECORD,,