deep-river 0.2.7__tar.gz → 0.2.8__tar.gz

{deep_river-0.2.7 → deep_river-0.2.8}/PKG-INFO

@@ -1,120 +1,34 @@
 Metadata-Version: 2.1
 Name: deep_river
-Version: 0.2.7
+Version: 0.2.8
 Summary: Online Deep Learning for river
-Home-page: https://online-ml.github.io/deep-river/
+Home-page: https://github.com/online-ml/deep-river
+License: BSD-3-Clause
+Keywords: online-learning,deep-learning,data-streams
 Author: Cedric Kulbach
 Author-email: cedric.kulbach@googlemail.com
-License: BSD-3
+Requires-Python: >=3.10,<3.13
 Classifier: License :: OSI Approved :: BSD License
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
-Requires-Python: >=3.6.0
+Requires-Dist: mkdocstrings[python] (>=0.19.0)
+Requires-Dist: numpy (>=1.26.4,<1.27.0)
+Requires-Dist: ordered-set (>=4.1.0,<4.2.0)
+Requires-Dist: pandas (>=2.2.2,<2.3.0)
+Requires-Dist: pytkdocs[numpy-style] (>=0.5.0)
+Requires-Dist: river
+Requires-Dist: scikit-learn (>=1.5.0,<1.6.0)
+Requires-Dist: torch (>=2.2.2,<2.3.0)
+Requires-Dist: torchviz (>=0.0.2,<0.1.0)
+Requires-Dist: tqdm (>=4.66.5,<4.67.0)
+Project-URL: Documentation, https://github.com/online-ml/deep-river
+Project-URL: Repository, https://github.com/online-ml/deep-river
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: scikit-learn~=1.5.0
-Requires-Dist: torch~=2.2.2
-Requires-Dist: pandas~=2.2.2
-Requires-Dist: numpy~=1.26.4
-Requires-Dist: river~=0.21.2
-Requires-Dist: scipy~=1.13.0
-Requires-Dist: tqdm~=4.66.5
-Requires-Dist: ordered-set~=4.1.0
-Requires-Dist: torchviz~=0.0.2
-Provides-Extra: dev
-Requires-Dist: scikit-learn~=1.5.0; extra == "dev"
-Requires-Dist: torch~=2.2.2; extra == "dev"
-Requires-Dist: pandas~=2.2.2; extra == "dev"
-Requires-Dist: numpy~=1.26.4; extra == "dev"
-Requires-Dist: river~=0.21.2; extra == "dev"
-Requires-Dist: scipy~=1.13.0; extra == "dev"
-Requires-Dist: tqdm~=4.66.5; extra == "dev"
-Requires-Dist: ordered-set~=4.1.0; extra == "dev"
-Requires-Dist: torchviz~=0.0.2; extra == "dev"
-Requires-Dist: graphviz>=0.20.3; extra == "dev"
-Requires-Dist: matplotlib>=3.9.2; extra == "dev"
-Requires-Dist: mypy>=1.11.1; extra == "dev"
-Requires-Dist: pre-commit>=3.8.0; extra == "dev"
-Requires-Dist: pytest>=8.3.2; extra == "dev"
-Requires-Dist: pytest-cov>=5.0.0; extra == "dev"
-Requires-Dist: black>=24.8.0; extra == "dev"
-Requires-Dist: flake8>=7.1.1; extra == "dev"
-Requires-Dist: isort>=5.13.2; extra == "dev"
-Requires-Dist: jupyter>=1.0.0; extra == "dev"
-Requires-Dist: pyupgrade==3.17.0; extra == "dev"
-Provides-Extra: docs
-Requires-Dist: scikit-learn~=1.5.0; extra == "docs"
-Requires-Dist: torch~=2.2.2; extra == "docs"
-Requires-Dist: pandas~=2.2.2; extra == "docs"
-Requires-Dist: numpy~=1.26.4; extra == "docs"
-Requires-Dist: river~=0.21.2; extra == "docs"
-Requires-Dist: scipy~=1.13.0; extra == "docs"
-Requires-Dist: tqdm~=4.66.5; extra == "docs"
-Requires-Dist: ordered-set~=4.1.0; extra == "docs"
-Requires-Dist: torchviz~=0.0.2; extra == "docs"
-Requires-Dist: flask>=2.0.2; extra == "docs"
-Requires-Dist: ipykernel>=6.9.0; extra == "docs"
-Requires-Dist: mike>=0.5.3; extra == "docs"
-Requires-Dist: mkdocs>=1.2.3; extra == "docs"
-Requires-Dist: mkdocs-awesome-pages-plugin>=2.7.0; extra == "docs"
-Requires-Dist: mkdocs-gen-files>=0.3.5; extra == "docs"
-Requires-Dist: mkdocs-charts-plugin>=0.0.8; extra == "docs"
-Requires-Dist: mkdocs-literate-nav>=0.4.1; extra == "docs"
-Requires-Dist: mkdocs-material>=8.1.11; extra == "docs"
-Requires-Dist: mkdocstrings[python]>=0.19.0; extra == "docs"
-Requires-Dist: pytkdocs[numpy-style]>=0.5.0; extra == "docs"
-Requires-Dist: ipython_genutils>=0.1.0; extra == "docs"
-Requires-Dist: mkdocs-jupyter>=0.20.0; extra == "docs"
-Requires-Dist: nbconvert>=6.4.2; extra == "docs"
-Requires-Dist: numpydoc>=1.2; extra == "docs"
-Requires-Dist: spacy>=3.2.2; extra == "docs"
-Requires-Dist: jinja2>=3.0.3; extra == "docs"
-Requires-Dist: dominate; extra == "docs"
-Requires-Dist: jupyter-client; extra == "docs"
-Requires-Dist: mkdocs-charts-plugin; extra == "docs"
-Requires-Dist: python-slugify; extra == "docs"
-Requires-Dist: watermark==2.3.1; extra == "docs"
-Provides-Extra: all
-Requires-Dist: graphviz>=0.20.3; extra == "all"
-Requires-Dist: matplotlib>=3.9.2; extra == "all"
-Requires-Dist: mypy>=1.11.1; extra == "all"
-Requires-Dist: pre-commit>=3.8.0; extra == "all"
-Requires-Dist: pytest>=8.3.2; extra == "all"
-Requires-Dist: pytest-cov>=5.0.0; extra == "all"
-Requires-Dist: black>=24.8.0; extra == "all"
-Requires-Dist: flake8>=7.1.1; extra == "all"
-Requires-Dist: isort>=5.13.2; extra == "all"
-Requires-Dist: jupyter>=1.0.0; extra == "all"
-Requires-Dist: pyupgrade==3.17.0; extra == "all"
-Requires-Dist: flask>=2.0.2; extra == "all"
-Requires-Dist: ipykernel>=6.9.0; extra == "all"
-Requires-Dist: mike>=0.5.3; extra == "all"
-Requires-Dist: mkdocs>=1.2.3; extra == "all"
-Requires-Dist: mkdocs-awesome-pages-plugin>=2.7.0; extra == "all"
-Requires-Dist: mkdocs-gen-files>=0.3.5; extra == "all"
-Requires-Dist: mkdocs-charts-plugin>=0.0.8; extra == "all"
-Requires-Dist: mkdocs-literate-nav>=0.4.1; extra == "all"
-Requires-Dist: mkdocs-material>=8.1.11; extra == "all"
-Requires-Dist: mkdocstrings[python]>=0.19.0; extra == "all"
-Requires-Dist: pytkdocs[numpy-style]>=0.5.0; extra == "all"
-Requires-Dist: ipython_genutils>=0.1.0; extra == "all"
-Requires-Dist: mkdocs-jupyter>=0.20.0; extra == "all"
-Requires-Dist: nbconvert>=6.4.2; extra == "all"
-Requires-Dist: numpydoc>=1.2; extra == "all"
-Requires-Dist: spacy>=3.2.2; extra == "all"
-Requires-Dist: jinja2>=3.0.3; extra == "all"
-Requires-Dist: dominate; extra == "all"
-Requires-Dist: jupyter-client; extra == "all"
-Requires-Dist: mkdocs-charts-plugin; extra == "all"
-Requires-Dist: python-slugify; extra == "all"
-Requires-Dist: watermark==2.3.1; extra == "all"
-Requires-Dist: dataclasses; python_version == "3.6"
-
 
 <p align="center">
   <img height="150px" src="https://raw.githubusercontent.com/online-ml/deep-river/master/docs/img/logo.png" alt="incremental dl logo">
@@ -124,6 +38,7 @@ Requires-Dist: dataclasses; python_version == "3.6"
     <a href="https://codecov.io/gh/online-ml/deep-river" > 
         <img src="https://codecov.io/gh/online-ml/deep-river/branch/master/graph/badge.svg?token=ZKUIISZAYA"/> 
     </a>
+    <img alt="PyPI - Python Version" src="https://img.shields.io/pypi/pyversions/deep-river">
     <img alt="PyPI - Downloads" src="https://img.shields.io/pypi/dm/deep-river">
     <img alt="GitHub" src="https://img.shields.io/github/license/online-ml/deep-river">
     <a href="https://joss.theoj.org/papers/6a76784f55e8b041d71a7fa776eb386a"><img src="https://joss.theoj.org/papers/6a76784f55e8b041d71a7fa776eb386a/status.svg"></a>
@@ -135,7 +50,9 @@ Requires-Dist: dataclasses; python_version == "3.6"
 </p>
 
 ## 📚 [Documentation](https://online-ml.github.io/deep-river/)
-The [documentation](https://online-ml.github.io/deep-river/) contains an overview of all features of this repository as well as the repository's full features list. In each of these, the git repo reference is listed in a section that shows examples of the features and functionality.
+The [documentation](https://online-ml.github.io/deep-river/) contains an overview of all features of this repository as well as the repository's full features list. 
+In each of these, the git repo reference is listed in a section that shows [examples](https://github.com/online-ml/deep-river/blob/master/docs/examples) of the features and functionality.
+As we are always looking for further use cases and examples, feel free to contribute to the documentation or the repository itself via a pull request
 
 ## 💈 Installation
 
@@ -250,7 +167,7 @@ MicroAverage(MAE): 34.31
 >>> from river.preprocessing import MinMaxScaler
 
 >>> dataset = CreditCard().take(5000)
->>> metric = metrics.ROCAUC(n_thresholds=50)
+>>> metric = metrics.RollingROCAUC(window_size=5000)
 
 >>> class MyAutoEncoder(nn.Module):
 ...     def __init__(self, n_features, latent_dim=3):
@@ -275,8 +192,8 @@ MicroAverage(MAE): 34.31
 ...     model.learn_one(x=x)
 ...     metric.update(y, score)
 ...
->>> print(f"ROCAUC: {metric.get():.4f}")
-ROCAUC: 0.7812
+>>> print(f"Rolling ROCAUC: {metric.get():.4f}")
+Rolling ROCAUC: 0.8901
 
 ```
 
@@ -289,3 +206,4 @@ ROCAUC: 0.7812
 <p align="center">
     <img src="https://lieferbotnet.de/wp-content/uploads/2022/09/LieferBotNet-Logo.png?raw=true" alt="Lieferbot net" height="200"/>
 </p>
+

{deep_river-0.2.7 → deep_river-0.2.8}/README.md

@@ -6,6 +6,7 @@
     <a href="https://codecov.io/gh/online-ml/deep-river" > 
         <img src="https://codecov.io/gh/online-ml/deep-river/branch/master/graph/badge.svg?token=ZKUIISZAYA"/> 
     </a>
+    <img alt="PyPI - Python Version" src="https://img.shields.io/pypi/pyversions/deep-river">
     <img alt="PyPI - Downloads" src="https://img.shields.io/pypi/dm/deep-river">
     <img alt="GitHub" src="https://img.shields.io/github/license/online-ml/deep-river">
     <a href="https://joss.theoj.org/papers/6a76784f55e8b041d71a7fa776eb386a"><img src="https://joss.theoj.org/papers/6a76784f55e8b041d71a7fa776eb386a/status.svg"></a>
@@ -17,7 +18,9 @@
 </p>
 
 ## 📚 [Documentation](https://online-ml.github.io/deep-river/)
-The [documentation](https://online-ml.github.io/deep-river/) contains an overview of all features of this repository as well as the repository's full features list. In each of these, the git repo reference is listed in a section that shows examples of the features and functionality.
+The [documentation](https://online-ml.github.io/deep-river/) contains an overview of all features of this repository as well as the repository's full features list. 
+In each of these, the git repo reference is listed in a section that shows [examples](https://github.com/online-ml/deep-river/blob/master/docs/examples) of the features and functionality.
+As we are always looking for further use cases and examples, feel free to contribute to the documentation or the repository itself via a pull request
 
 ## 💈 Installation
 
@@ -132,7 +135,7 @@ MicroAverage(MAE): 34.31
 >>> from river.preprocessing import MinMaxScaler
 
 >>> dataset = CreditCard().take(5000)
->>> metric = metrics.ROCAUC(n_thresholds=50)
+>>> metric = metrics.RollingROCAUC(window_size=5000)
 
 >>> class MyAutoEncoder(nn.Module):
 ...     def __init__(self, n_features, latent_dim=3):
@@ -157,8 +160,8 @@ MicroAverage(MAE): 34.31
 ...     model.learn_one(x=x)
 ...     metric.update(y, score)
 ...
->>> print(f"ROCAUC: {metric.get():.4f}")
-ROCAUC: 0.7812
+>>> print(f"Rolling ROCAUC: {metric.get():.4f}")
+Rolling ROCAUC: 0.8901
 
 ```
 

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/__version__.py

@@ -1,3 +1,3 @@
-VERSION = (0, 2, 7)
+VERSION = (0, 2, 8)
 
 __version__ = ".".join(map(str, VERSION))  # noqa: F401

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/anomaly/__init__.py

@@ -1,11 +1,7 @@
 from .ae import Autoencoder
 from .probability_weighted_ae import ProbabilityWeightedAutoencoder
 from .rolling_ae import RollingAutoencoder
-from .scaler import (
-    AnomalyMeanScaler,
-    AnomalyMinMaxScaler,
-    AnomalyStandardScaler,
-)
+from .scaler import AnomalyMeanScaler, AnomalyMinMaxScaler, AnomalyStandardScaler
 
 """
 This module contains the anomaly detection algorithms for the

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/anomaly/ae.py

@@ -68,7 +68,7 @@ class Autoencoder(DeepEstimator, AnomalyDetector):
     >>> from river.preprocessing import MinMaxScaler
 
     >>> dataset = CreditCard().take(5000)
-    >>> metric = metrics.ROCAUC(n_thresholds=50)
+    >>> metric = metrics.RollingROCAUC(window_size=5000)
 
     >>> class MyAutoEncoder(torch.nn.Module):
     ...     def __init__(self, n_features, latent_dim=3):
@@ -93,8 +93,8 @@ class Autoencoder(DeepEstimator, AnomalyDetector):
     ...    model.learn_one(x=x)
     ...    metric.update(y, score)
     ...
-    >>> print(f"ROCAUC: {metric.get():.4f}")
-    ROCAUC: 0.7812
+    >>> print(f"Rolling ROCAUC: {metric.get():.4f}")
+    Rolling ROCAUC: 0.8901
     """
 
     def __init__(
@@ -212,9 +212,7 @@ class Autoencoder(DeepEstimator, AnomalyDetector):
 
         self._adapt_input_dim(x)
 
-        x_t = dict2tensor(
-            x, features=self.observed_features, device=self.device
-        )
+        x_t = dict2tensor(x, features=self.observed_features, device=self.device)
         self.module.eval()
         with torch.inference_mode():
             x_pred = self.module(x_t)

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/anomaly/probability_weighted_ae.py

@@ -62,7 +62,7 @@ class ProbabilityWeightedAutoencoder(ae.Autoencoder):
 
     >>> _ = manual_seed(42)
     >>> dataset = CreditCard().take(5000)
-    >>> metric = metrics.ROCAUC(n_thresholds=50)
+    >>> metric = metrics.RollingROCAUC(window_size=5000)
 
     >>> class MyAutoEncoder(torch.nn.Module):
     ...     def __init__(self, n_features, latent_dim=3):
@@ -87,8 +87,8 @@ class ProbabilityWeightedAutoencoder(ae.Autoencoder):
     ...    model.learn_one(x=x)
     ...    metric.update(y, score)
     ...
-    >>> print(f"ROCAUC: {metric.get():.4f}")
-    ROCAUC: 0.7572
+    >>> print(f"Rolling ROCAUC: {metric.get():.4f}")
+    Rolling ROCAUC: 0.8530
     """
 
     def __init__(
@@ -114,9 +114,7 @@ class ProbabilityWeightedAutoencoder(ae.Autoencoder):
         )
         self.window_size = window_size
         self.skip_threshold = skip_threshold
-        self.rolling_mean = utils.Rolling(
-            stats.Mean(), window_size=window_size
-        )
+        self.rolling_mean = utils.Rolling(stats.Mean(), window_size=window_size)
         self.rolling_var = utils.Rolling(stats.Var(), window_size=window_size)
 
     def learn_one(
@@ -165,10 +163,7 @@ class ProbabilityWeightedAutoencoder(ae.Autoencoder):
 
         loss_scaled = (losses_numpy - mean) / math.sqrt(var)
         prob = ndtr(loss_scaled)
-        loss = (
-            torch.tensor((self.skip_threshold - prob) / self.skip_threshold)
-            * loss
-        )
+        loss = torch.tensor((self.skip_threshold - prob) / self.skip_threshold) * loss
 
         self.optimizer.zero_grad()
         loss.backward()

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/anomaly/rolling_ae.py

@@ -11,9 +11,7 @@ from deep_river.utils.tensor_conversion import deque2rolling_tensor
 
 
 class _TestLSTMAutoencoder(nn.Module):
-    def __init__(
-        self, n_features, hidden_size=30, n_layers=1, batch_first=False
-    ):
+    def __init__(self, n_features, hidden_size=30, n_layers=1, batch_first=False):
         super().__init__()
         self.n_features = n_features
         self.hidden_size = hidden_size
@@ -142,9 +140,7 @@ class RollingAutoencoder(RollingDeepEstimator, anomaly.base.AnomalyDetector):
         loss.backward()
         self.optimizer.step()
 
-    def learn_one(
-        self, x: dict, y: Any = None, **kwargs
-    ) -> "RollingAutoencoder":
+    def learn_one(self, x: dict, y: Any = None, **kwargs) -> "RollingAutoencoder":
         """
         Performs one step of training with a single example.
 

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/anomaly/scaler.py

@@ -145,9 +145,7 @@ class AnomalyStandardScaler(AnomalyScaler):
         super().__init__(anomaly_detector)
         self.rolling = rolling
         self.window_size = window_size
-        self.mean = (
-            utils.Rolling(Mean(), self.window_size) if self.rolling else Mean()
-        )
+        self.mean = utils.Rolling(Mean(), self.window_size) if self.rolling else Mean()
         self.sq_mean = (
             utils.Rolling(Mean(), self.window_size) if self.rolling else Mean()
         )
@@ -208,9 +206,7 @@ class AnomalyMeanScaler(AnomalyScaler):
         super().__init__(anomaly_detector=anomaly_detector)
         self.rolling = rolling
         self.window_size = window_size
-        self.mean = (
-            utils.Rolling(Mean(), self.window_size) if self.rolling else Mean()
-        )
+        self.mean = utils.Rolling(Mean(), self.window_size) if self.rolling else Mean()
 
     def score_one(self, *args):
         """

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/base.py

@@ -21,9 +21,7 @@ try:
     from graphviz import Digraph
     from torchviz import make_dot
 except ImportError as e:
-    raise ValueError(
-        "You have to install graphviz to use the draw method"
-    ) from e
+    raise ValueError("You have to install graphviz to use the draw method") from e
 
 
 class DeepEstimator(base.Estimator):
@@ -119,9 +117,7 @@ class DeepEstimator(base.Estimator):
         first_parameter = next(self.module.parameters())
         input_shape = first_parameter.size()
         y_pred = self.module(torch.rand(input_shape))
-        return make_dot(
-            y_pred.mean(), params=dict(self.module.named_parameters())
-        )
+        return make_dot(y_pred.mean(), params=dict(self.module.named_parameters()))
 
     def initialize_module(self, x: dict | pd.DataFrame, **kwargs):
         """
@@ -151,9 +147,7 @@ class DeepEstimator(base.Estimator):
             )
 
         self.module.to(self.device)
-        self.optimizer = self.optimizer_func(
-            self.module.parameters(), lr=self.lr
-        )
+        self.optimizer = self.optimizer_func(self.module.parameters(), lr=self.lr)
         self.module_initialized = True
 
         self._get_input_output_layers(n_features=n_features)
@@ -245,9 +239,7 @@ class DeepEstimator(base.Estimator):
                 tracker.ordered_modules[0], SUPPORTED_LAYERS
             ):
                 self.input_layer = tracker.ordered_modules[0]
-                self.input_expansion_instructions = load_instructions(
-                    self.input_layer
-                )
+                self.input_expansion_instructions = load_instructions(self.input_layer)
             else:
                 warnings.warn(
                     "The model will not be able to adapt its input layer to "

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/classification/__init__.py

@@ -1,9 +1,6 @@
 from deep_river.classification.classifier import Classifier
 from deep_river.classification.rolling_classifier import RollingClassifier
-from deep_river.classification.zoo import (
-    LogisticRegression,
-    MultiLayerPerceptron,
-)
+from deep_river.classification.zoo import LogisticRegression, MultiLayerPerceptron
 
 """
 This module contains the classifiers for the deep_river package.

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/classification/classifier.py

@@ -227,9 +227,7 @@ class Classifier(DeepEstimator, base.MiniBatchClassifier):
         self._adapt_input_dim(x)
         self._adapt_output_dim(y)
 
-        x_t = dict2tensor(
-            x, features=self.observed_features, device=self.device
-        )
+        x_t = dict2tensor(x, features=self.observed_features, device=self.device)
 
         return self._learn(x=x_t, y=y)
 
@@ -254,16 +252,12 @@ class Classifier(DeepEstimator, base.MiniBatchClassifier):
 
         self._adapt_input_dim(x)
 
-        x_t = dict2tensor(
-            x, features=self.observed_features, device=self.device
-        )
+        x_t = dict2tensor(x, features=self.observed_features, device=self.device)
 
         self.module.eval()
         with torch.inference_mode():
             y_pred = self.module(x_t)
-        return output2proba(
-            y_pred, self.observed_classes, self.output_is_logit
-        )[0]
+        return output2proba(y_pred, self.observed_classes, self.output_is_logit)[0]
 
     def _update_observed_classes(self, y) -> bool:
         n_existing_classes = len(self.observed_classes)
@@ -292,13 +286,13 @@ class Classifier(DeepEstimator, base.MiniBatchClassifier):
                 output=True,
             )
 
-    def learn_many(self, x: pd.DataFrame, y: pd.Series) -> "Classifier":
+    def learn_many(self, X: pd.DataFrame, y: pd.Series) -> "Classifier":
         """
         Performs one step of training with a batch of examples.
 
         Parameters
         ----------
-        x
+        X
             Input examples.
         y
             Target values.
@@ -311,14 +305,14 @@ class Classifier(DeepEstimator, base.MiniBatchClassifier):
         # check if model is initialized
 
         if not self.module_initialized:
-            self._update_observed_features(x)
+            self._update_observed_features(X)
             self._update_observed_classes(y)
-            self.initialize_module(x=x, **self.kwargs)
+            self.initialize_module(x=X, **self.kwargs)
 
-        self._adapt_input_dim(x)
+        self._adapt_input_dim(X)
         self._adapt_output_dim(y)
 
-        x_t = df2tensor(x, features=self.observed_features, device=self.device)
+        x_t = df2tensor(X, features=self.observed_features, device=self.device)
 
         return self._learn(x=x_t, y=y)
 

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/classification/rolling_classifier.py

@@ -6,10 +6,7 @@ from river.base.typing import ClfTarget
 
 from deep_river.base import RollingDeepEstimator
 from deep_river.classification import Classifier
-from deep_river.utils.tensor_conversion import (
-    deque2rolling_tensor,
-    output2proba,
-)
+from deep_river.utils.tensor_conversion import deque2rolling_tensor, output2proba
 
 
 class _TestLSTM(torch.nn.Module):
@@ -150,9 +147,7 @@ class RollingClassifier(Classifier, RollingDeepEstimator):
             "check_predict_proba_one",
         }
 
-    def learn_one(
-        self, x: dict, y: ClfTarget, **kwargs
-    ) -> "RollingClassifier":
+    def learn_one(self, x: dict, y: ClfTarget, **kwargs) -> "RollingClassifier":
         """
         Performs one step of training with the most recent training examples
         stored in the sliding window.
@@ -177,9 +172,7 @@ class RollingClassifier(Classifier, RollingDeepEstimator):
 
         self._adapt_input_dim(x)
         self._adapt_output_dim(y)
-        self._x_window.append(
-            [x.get(feature, 0) for feature in self.observed_features]
-        )
+        self._x_window.append([x.get(feature, 0) for feature in self.observed_features])
 
         # training process
         if len(self._x_window) == self.window_size:
@@ -217,20 +210,18 @@ class RollingClassifier(Classifier, RollingDeepEstimator):
         with torch.inference_mode():
             x_t = deque2rolling_tensor(x_win, device=self.device)
             y_pred = self.module(x_t)
-            proba = output2proba(
-                y_pred, self.observed_classes, self.output_is_logit
-            )
+            proba = output2proba(y_pred, self.observed_classes, self.output_is_logit)
 
         return proba[0]
 
-    def learn_many(self, x: pd.DataFrame, y: pd.Series) -> "RollingClassifier":
+    def learn_many(self, X: pd.DataFrame, y: pd.Series) -> "RollingClassifier":
         """
         Performs one step of training with the most recent training examples
         stored in the sliding window.
 
         Parameters
         ----------
-        x
+        X
             Input examples.
         y
             Target values.
@@ -243,13 +234,13 @@ class RollingClassifier(Classifier, RollingDeepEstimator):
         # check if model is initialized
         if not self.module_initialized:
             self._update_observed_classes(y)
-            self._update_observed_features(x)
-            self.initialize_module(x=x, **self.kwargs)
+            self._update_observed_features(X)
+            self.initialize_module(x=X, **self.kwargs)
 
-        self._adapt_input_dim(x)
+        self._adapt_input_dim(X)
         self._adapt_output_dim(y)
-        x = x[list(self.observed_features)]
-        self._x_window.extend(x.values.tolist())
+        X = X[list(self.observed_features)]
+        self._x_window.extend(X.values.tolist())
 
         if self.is_class_incremental:
             self._adapt_output_dim(y)

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/classification/zoo.py

@@ -205,9 +205,7 @@ class MultiLayerPerceptron(Classifier):
         def __init__(self, n_width, n_layers, n_features):
             super().__init__()
             hidden = [nn.Linear(n_features, n_width)]
-            hidden += [
-                nn.Linear(n_width, n_width) for _ in range(n_layers - 1)
-            ]
+            hidden += [nn.Linear(n_width, n_width) for _ in range(n_layers - 1)]
             self.hidden = nn.ModuleList(hidden)
             self.denselast = nn.Linear(n_width, 1)
 

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/regression/multioutput.py

@@ -106,9 +106,7 @@ class MultiTargetRegressor(RiverMultiTargetRegressor, DeepEstimator):
             seed=seed,
             **kwargs,
         )
-        self.observed_targets: OrderedDict[FeatureName, RegTarget] = (
-            OrderedDict()
-        )
+        self.observed_targets: OrderedDict[FeatureName, RegTarget] = OrderedDict()
 
     @classmethod
     def _unit_test_params(cls):
@@ -195,7 +193,5 @@ class MultiTargetRegressor(RiverMultiTargetRegressor, DeepEstimator):
         self.module.eval()
         with torch.inference_mode():
             y_pred_t = self.module(x_t).squeeze().tolist()
-            y_pred = {
-                t: y_pred_t[i] for i, t in enumerate(self.observed_targets)
-            }
+            y_pred = {t: y_pred_t[i] for i, t in enumerate(self.observed_targets)}
         return y_pred

{deep_river-0.2.7 → deep_river-0.2.8}/deep_river/regression/regressor.py

@@ -1,4 +1,4 @@
-from typing import Callable, List, Type, Union
+from typing import Callable, Type, Union
 
 import pandas as pd
 import torch
@@ -6,11 +6,7 @@ from river import base
 from river.base.typing import RegTarget
 
 from deep_river.base import DeepEstimator
-from deep_river.utils.tensor_conversion import (
-    df2tensor,
-    dict2tensor,
-    float2tensor,
-)
+from deep_river.utils.tensor_conversion import df2tensor, dict2tensor, float2tensor
 
 
 class _TestModule(torch.nn.Module):
@@ -143,9 +139,7 @@ class Regressor(DeepEstimator, base.MiniBatchRegressor):
             self._update_observed_features(x)
             self.initialize_module(x=x, **self.kwargs)
         self._adapt_input_dim(x)
-        x_t = dict2tensor(
-            x, features=self.observed_features, device=self.device
-        )
+        x_t = dict2tensor(x, features=self.observed_features, device=self.device)
         y_t = float2tensor(y, device=self.device)
 
         self._learn(x_t, y_t)
@@ -177,9 +171,7 @@ class Regressor(DeepEstimator, base.MiniBatchRegressor):
             self._update_observed_features(x)
             self.initialize_module(x=x, **self.kwargs)
         self._adapt_input_dim(x)
-        x_t = dict2tensor(
-            x, features=self.observed_features, device=self.device
-        )
+        x_t = dict2tensor(x, features=self.observed_features, device=self.device)
 
         self.module.eval()
         with torch.inference_mode():
@@ -209,14 +201,12 @@ class Regressor(DeepEstimator, base.MiniBatchRegressor):
 
         self._adapt_input_dim(X)
         X_t = df2tensor(X, features=self.observed_features, device=self.device)
-        y_t = torch.tensor(
-            y, device=self.device, dtype=torch.float32
-        ).unsqueeze(1)
+        y_t = torch.tensor(y, device=self.device, dtype=torch.float32).unsqueeze(1)
 
         self._learn(X_t, y_t)
         return self
 
-    def predict_many(self, X: pd.DataFrame) -> List:
+    def predict_many(self, X: pd.DataFrame) -> pd.Series:
         """
         Predicts the target value for a batch of examples.