aisp 0.1.40__py3-none-any.whl → 0.1.41__py3-none-any.whl

aisp/base/_classifier.py

@@ -1,10 +1,11 @@
 """Base class for classification algorithm."""
 
 from abc import ABC, abstractmethod
-from typing import Optional
+from typing import Optional, Union
 
 import numpy.typing as npt
 
+from ..utils import slice_index_list_by_class
 from ..utils.metrics import accuracy_score
 
 
@@ -14,6 +15,8 @@ class BaseClassifier(ABC):
     and implementing the ``get_params`` method.
     """
 
+    classes: Optional[Union[npt.NDArray, list]] = None
+
     @abstractmethod
     def fit(self, X: npt.NDArray, y: npt.NDArray, verbose: bool = True):
         """
@@ -77,6 +80,23 @@ class BaseClassifier(ABC):
         y_pred = self.predict(X)
         return accuracy_score(y, y_pred)
 
+    def _slice_index_list_by_class(self, y: npt.NDArray) -> dict:
+        """
+        The function ``_slice_index_list_by_class(...)``, separates the indices of the lines \
+        according to the output class, to loop through the sample array, only in positions where \
+        the output is the class being trained.
+
+        Parameters
+        ----------
+        * y (npt.NDArray): Receives a ``y``[``N sample``] array with the output classes of the \
+            ``X`` sample array.
+
+        returns
+        ----------
+        * dict: A dictionary with the list of array positions(``y``), with the classes as key.
+        """
+        return slice_index_list_by_class(self.classes, y)
+
     def get_params(self, deep: bool = True) -> dict:  # pylint: disable=W0613
         """
         The get_params function Returns a dictionary with the object's main parameters.

aisp/nsa/_negative_selection.py

@@ -12,7 +12,6 @@ from ._ns_core import (
     check_detector_rnsa_validity,
 )
 from ..exceptions import MaxDiscardsReachedError
-from ..utils import slice_index_list_by_class
 from ..utils.distance import (
     min_distance_to_class_vectors,
     get_metric_code,
@@ -111,6 +110,11 @@ class RNSA(BaseNSA):
         self._detectors: Union[dict, None] = None
         self.classes: npt.NDArray = None
 
+    @property
+    def detectors(self) -> Dict[str, list[Detector]]:
+        """Returns the trained detectors, organized by class."""
+        return self._detectors
+
     def fit(self, X: npt.NDArray, y: npt.NDArray, verbose: bool = True):
         """
         The function ``fit(...)``, performs the training according to ``X`` and ``y``, using the
@@ -141,7 +145,7 @@ class RNSA(BaseNSA):
         # Dictionary that will store detectors with classes as keys.
         list_detectors_by_class = {}
         # Separates the classes for training.
-        sample_index = self.__slice_index_list_by_class(y)
+        sample_index = self._slice_index_list_by_class(y)
         # Progress bar for generating all detectors.
         if verbose:
             progress = tqdm(
@@ -168,7 +172,9 @@ class RNSA(BaseNSA):
                 # If the detector is valid, add it to the list of valid detectors.
                 if valid_detector is not False:
                     discard_count = 0
-                    radius = valid_detector[1] if self.algorithm == "V-detector" else None
+                    radius = (
+                        valid_detector[1] if self.algorithm == "V-detector" else None
+                    )
                     valid_detectors_set.append(Detector(vector_x, radius))
                     if verbose:
                         progress.update(1)
@@ -178,7 +184,7 @@ class RNSA(BaseNSA):
                         raise MaxDiscardsReachedError(_class_)
 
             # Add detectors, with classes as keys in the dictionary.
-            list_detectors_by_class[_class_] = np.array(valid_detectors_set)
+            list_detectors_by_class[_class_] = valid_detectors_set
         # Notify completion of detector generation for the classes.
         if verbose:
             progress.set_description(
@@ -253,24 +259,6 @@ class RNSA(BaseNSA):
                 c.append(max(average_distance, key=average_distance.get))
         return np.array(c)
 
-    def __slice_index_list_by_class(self, y: npt.NDArray) -> dict:
-        """
-        The function ``__slice_index_list_by_class(...)``, separates the indices of the lines
-        according to the output class, to loop through the sample array, only in positions where
-        the output is the class being trained.
-
-        Parameters
-        ----------
-        * y (npt.NDArray)
-            Receives a ``y``[``N sample``] array with the output classes of the \
-            ``X`` sample array.
-
-        Returns
-        ----------
-        * dict: A dictionary with the list of array positions(``y``), with the classes as key.
-        """
-        return slice_index_list_by_class(self.classes, y)
-
     def __checks_valid_detector(
         self, x_class: npt.NDArray = None, vector_x: npt.NDArray = None
     ) -> Union[bool, tuple[bool, float]]:
@@ -503,6 +491,11 @@ class BNSA(BaseNSA):
         self._detectors: Optional[dict] = None
         self._detectors_stack: npt.NDArray = None
 
+    @property
+    def detectors(self) -> Dict[str, npt.NDArray[np.bool_]]:
+        """Returns the trained detectors, organized by class."""
+        return self._detectors
+
     def fit(self, X: npt.NDArray, y: npt.NDArray, verbose: bool = True):
         """
         The function ``fit(...)``, performs the training according to ``X`` and ``y``, using the
@@ -531,7 +524,7 @@ class BNSA(BaseNSA):
         # Dictionary that will store detectors with classes as keys.
         list_detectors_by_class = {}
         # Separates the classes for training.
-        sample_index: dict = self.__slice_index_list_by_class(y)
+        sample_index: dict = self._slice_index_list_by_class(y)
         # Progress bar for generating all detectors.
         if verbose:
             progress = tqdm(
@@ -662,21 +655,3 @@ class BNSA(BaseNSA):
                 class_differences[_class_] = distances.sum() / self.N
 
         c.append(max(class_differences, key=class_differences.get))
-
-    def __slice_index_list_by_class(self, y: npt.NDArray) -> dict:
-        """
-        The function ``__slice_index_list_by_class(...)``, separates the indices of the lines
-        according to the output class, to loop through the sample array, only in positions where
-        the output is the class being trained.
-
-        Parameters
-        ----------
-        * y (``npt.NDArray``):
-            Receives a ``y``[``N sample``] array with the output classes of the ``X``
-            sample array.
-
-        Returns
-        ----------
-        * dict: A dictionary with the list of array positions(``y``), with the classes as key.
-        """
-        return slice_index_list_by_class(self.classes, y)

{aisp-0.1.40.dist-info → aisp-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aisp
-Version: 0.1.40
+Version: 0.1.41
 Summary: Package with techniques of artificial immune systems.
 Author-email: João Paulo da Silva Barros <jpsilvabarr@gmail.com>
 Maintainer-email: Alison Zille Lopes <alisonzille@gmail.com>

{aisp-0.1.40.dist-info → aisp-0.1.41.dist-info}/RECORD

@@ -1,18 +1,18 @@
 aisp/__init__.py,sha256=pKOdkhFxWup_lhQPofFTk4rpqd0_IlYygHnkBhE2g2w,111
 aisp/exceptions.py,sha256=ONiKCZrBhfUaMO1vYHnOr1_eL9PtVJ7b_HRroMbORpc,1405
 aisp/base/__init__.py,sha256=rpb9ADmBtNpRAEJw61c7CsbZhHA2vGQOvjFmpk7w39c,100
-aisp/base/_classifier.py,sha256=qv6TxzfnRwO7HvN7CA8PS1dueAnm-RifelRnCjrK9D0,3027
+aisp/base/_classifier.py,sha256=Dvimv7JVfqeg4ws9NqZPj0Ue_fNh9kEEXrspOrLdBoQ,3845
 aisp/nsa/__init__.py,sha256=3-fSIJCMB_jcLGTr6DcVkloRfWyQJ8-JzD0CLhvpvgc,411
 aisp/nsa/_base.py,sha256=cT404V5onduK19WW6ajxy_yUZha3Fk8KZJvJHt_UZBM,4250
-aisp/nsa/_negative_selection.py,sha256=0_VI3rAeVzPcSRwjMjqCq6RT-1_a_-HCFtIIJL6jpyc,29582
+aisp/nsa/_negative_selection.py,sha256=mWyjRmTMOJxtL8xHDooN3RreyLgq5JqmEb_CYAHBaCk,28485
 aisp/nsa/_ns_core.py,sha256=VWt_L2ODXtngJ7BVmSVT7i7kK98axibOmCzbYCkjeqI,4934
 aisp/utils/__init__.py,sha256=-r--qdfTq4dVJU-VoJ7dvXTNS5_I1q1aGAQ02zNBOGw,217
 aisp/utils/_multiclass.py,sha256=t0RLIdUoUbvFL30a8XfO3BMXc8MimtLCHNbGCDL2BQk,1051
 aisp/utils/distance.py,sha256=SZTE5KEB7QX9vKoY9GfFOfFGbswVhQM6-9baLq4ZXns,6304
 aisp/utils/metrics.py,sha256=o6aD6qTLQtXhWNOMGjgweHMQ6SMQ2hh42NcSEmgY3RI,1292
 aisp/utils/sanitizers.py,sha256=7wJSo5ZkRX1g16C9sEN8TWQ5l9Gzfl0-8CgLULnPNlk,1873
-aisp-0.1.40.dist-info/licenses/LICENSE,sha256=fTqV5eBpeAZO0_jit8j4Ref9ikBSlHJ8xwj5TLg7gFk,7817
-aisp-0.1.40.dist-info/METADATA,sha256=em8eJAx3lr87ZZ1ajsoSkIj-itauLz25E1FKcdjf1QU,4818
-aisp-0.1.40.dist-info/WHEEL,sha256=GHB6lJx2juba1wDgXDNlMTyM13ckjBMKf-OnwgKOCtA,91
-aisp-0.1.40.dist-info/top_level.txt,sha256=Q5aJi_rAVT5UNS1As0ZafoyS5dwNibnoyOYV7RWUB9s,5
-aisp-0.1.40.dist-info/RECORD,,
+aisp-0.1.41.dist-info/licenses/LICENSE,sha256=fTqV5eBpeAZO0_jit8j4Ref9ikBSlHJ8xwj5TLg7gFk,7817
+aisp-0.1.41.dist-info/METADATA,sha256=W0BxdV6IefnxXgDIe-eSa1XvGgFsqqsAnvIDxzL8EGE,4818
+aisp-0.1.41.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+aisp-0.1.41.dist-info/top_level.txt,sha256=Q5aJi_rAVT5UNS1As0ZafoyS5dwNibnoyOYV7RWUB9s,5
+aisp-0.1.41.dist-info/RECORD,,

{aisp-0.1.40.dist-info → aisp-0.1.41.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.0)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any