mlquantify 0.1.5__tar.gz → 0.1.6__tar.gz

{mlquantify-0.1.5/mlquantify.egg-info → mlquantify-0.1.6}/PKG-INFO

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

mlquantify-0.1.6/VERSION.txt

@@ -0,0 +1 @@
+0.1.6

{mlquantify-0.1.5 → mlquantify-0.1.6}/mlquantify/evaluation/protocol.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+from logging import warning
 import numpy as np
 from typing import Generator, Tuple
 from tqdm import tqdm
@@ -77,18 +78,11 @@ class Protocol(ABC):
         Generator[np.ndarray, np.ndarray]
             A generator that yields the indices for each split.
         """
-        indices = np.arange(X.shape[0])
-        for idx in self._split_indices_masks(X, y):
-            indexes = indices[idx]
-            yield indexes
-
-    def _split_indices_masks(self, X: np.ndarray, y: np.ndarray) -> Generator[Tuple[np.ndarray, np.ndarray]]:
         for idx in self._iter_indices(X, y):
+            if len(idx) > len(X):
+                warning(f"Batch size {len(idx)} exceeds dataset size {len(X)}. Replacement sampling will be used.")
+            yield idx
 
-            mask = np.zeros(X.shape[0], dtype=bool)
-            mask[idx] = True 
-
-            yield mask
 
     @abstractmethod
     def _iter_indices(self, X, y):

{mlquantify-0.1.5 → mlquantify-0.1.6/mlquantify.egg-info}/PKG-INFO

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

mlquantify-0.1.5/VERSION.txt

@@ -1 +0,0 @@
-0.1.5