chemotools 0.1.3__tar.gz → 0.1.4__tar.gz

{chemotools-0.1.3 → chemotools-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chemotools
-Version: 0.1.3
+Version: 0.1.4
 Summary: Package to integrate chemometrics in scikit-learn pipelines
 Home-page: https://github.com/paucablop/chemotools
 Author: Pau Cabaneros Lopez

{chemotools-0.1.3 → chemotools-0.1.4}/chemotools/augmentation/uniform_noise.py

@@ -11,10 +11,10 @@ class UniformNoise(OneToOneFeatureMixin, BaseEstimator, TransformerMixin):
 
     Parameters
     ----------
-    low : float, default=0.0
+    min : float, default=0.0
         The lower bound of the uniform distribution.
 
-    high : float, default=0.0
+    max : float, default=0.0
         The upper bound of the uniform distribution.
 
     random_state : int, default=None
@@ -38,9 +38,9 @@ class UniformNoise(OneToOneFeatureMixin, BaseEstimator, TransformerMixin):
     """
 
 
-    def __init__(self, low: float = 0.0, high: float = 0.0, random_state: int = None):
-        self.low = low
-        self.high = high
+    def __init__(self, min: float = 0.0, max: float = 0.0, random_state: int = None):
+        self.min = min
+        self.max = max
         self.random_state = random_state
 
     def fit(self, X: np.ndarray, y=None) -> "UniformNoise":
@@ -109,4 +109,4 @@ class UniformNoise(OneToOneFeatureMixin, BaseEstimator, TransformerMixin):
         return X_.reshape(-1, 1) if X_.ndim == 1 else X_
 
     def _add_random_noise(self, x) -> np.ndarray:
-        return x + self._rng.uniform(self.low, self.high, size=x.shape)
+        return x + self._rng.uniform(self.min, self.max, size=x.shape)

{chemotools-0.1.3 → chemotools-0.1.4}/chemotools/feature_selection/_range_cut.py

@@ -78,6 +78,7 @@ class RangeCut(BaseEstimator, SelectorMixin):
         else:
             self.start_index_ = self._find_index(self.start)
             self.end_index_ = self._find_index(self.end)
+            self.wavenumbers_ = self.wavenumbers[self.start_index_ : self.end_index_]
 
         return self
     

{chemotools-0.1.3 → chemotools-0.1.4}/chemotools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chemotools
-Version: 0.1.3
+Version: 0.1.4
 Summary: Package to integrate chemometrics in scikit-learn pipelines
 Home-page: https://github.com/paucablop/chemotools
 Author: Pau Cabaneros Lopez

{chemotools-0.1.3 → chemotools-0.1.4}/tests/test_functionality.py

@@ -622,6 +622,7 @@ def test_range_cut_by_wavenumber_with_list():
 
     # Assert
     assert np.allclose(spectrum_corrected[0], spectrum[0][1:7], atol=1e-8)
+    assert range_cut.wavenumbers_ == [2, 3, 4, 5, 6, 7]
 
 
 def test_range_cut_by_wavenumber_with_dataframe():
@@ -740,7 +741,7 @@ def test_subtract_reference_without_reference(spectrum):
 def test_uniform_noise():
     # Arrange
     spectrum = np.ones(10000).reshape(1, -1)
-    uniform_noise = UniformNoise(low=-1, high=1, random_state=42)
+    uniform_noise = UniformNoise(min=-1, max=1, random_state=42)
 
     # Act
     spectrum_corrected = uniform_noise.fit_transform(spectrum)