deeplotx 0.3.1__tar.gz → 0.4.1__tar.gz

{deeplotx-0.3.1 → deeplotx-0.4.1}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.3.1
+Version: 0.4.1
 Summary: Easy-2-use long text classifier trainers.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: huggingface-hub[hf-xet]>=0.30.2
+Requires-Dist: hf-xet>=1.0.5
 Requires-Dist: jupyter>=1.1.1
 Requires-Dist: numpy>=2.2.5
 Requires-Dist: python-dotenv>=1.1.0

{deeplotx-0.3.1 → deeplotx-0.4.1}/deeplotx/nn/__init__.py

@@ -1,3 +1,5 @@
 from .linear_regression import LinearRegression
 from .logistic_regression import LogisticRegression
 from .softmax_regression import SoftmaxRegression
+from .recursive_sequential import RecursiveSequential
+from .auto_regression import AutoRegression

deeplotx-0.4.1/deeplotx/nn/auto_regression.py

@@ -0,0 +1,9 @@
+from deeplotx.nn import RecursiveSequential
+
+
+class AutoRegression(RecursiveSequential):
+    def __init__(self, feature_dim: int, hidden_dim: int | None = None,
+                 recursive_layers: int = 2, model_name: str | None = None):
+        super().__init__(input_dim=feature_dim, output_dim=feature_dim,
+                         hidden_dim=hidden_dim, recursive_layers=recursive_layers,
+                         model_name=model_name)

{deeplotx-0.3.1 → deeplotx-0.4.1}/deeplotx/nn/base_neural_network.py

@@ -31,7 +31,7 @@ class BaseNeuralNetwork(nn.Module):
         return alpha * (rho * self.l1(_lambda=1.) + (1 - rho) * self.l2(_lambda=1.))
 
     @abstractmethod
-    def forward(self, x) -> torch.Tensor: ...
+    def forward(self, *args, **kwargs) -> torch.Tensor: ...
 
     def predict(self, x) -> torch.Tensor:
         __train = self.training

deeplotx-0.4.1/deeplotx/nn/recursive_sequential.py

@@ -0,0 +1,30 @@
+from typing_extensions import override
+
+import torch
+from torch import nn
+
+from deeplotx.nn.base_neural_network import BaseNeuralNetwork
+from deeplotx.nn import LinearRegression
+
+
+class RecursiveSequential(BaseNeuralNetwork):
+    def __init__(self, input_dim: int, output_dim: int,
+                 hidden_dim: int | None = None, recursive_layers: int = 2,
+                 model_name: str | None = None):
+        super().__init__(model_name=model_name)
+        if hidden_dim is None:
+            hidden_dim = input_dim
+        self.lstm = nn.LSTM(input_size=input_dim, hidden_size=hidden_dim,
+                            num_layers=recursive_layers, batch_first=True,
+                            bias=True, bidirectional=True)
+        self.regressive_head = LinearRegression(input_dim=hidden_dim * 2, output_dim=output_dim)
+
+    def initial_state(self, batch_size: int = 1) -> tuple[torch.Tensor, torch.Tensor]:
+        zeros = torch.zeros(self.lstm.num_layers * 2, batch_size, self.lstm.hidden_size)
+        return zeros, zeros
+
+    @override
+    def forward(self, x, state: tuple[torch.Tensor, torch.Tensor]) -> tuple[torch.Tensor, tuple[torch.Tensor, torch.Tensor]]:
+        x, (hidden_state, cell_state) = self.lstm(x, state)
+        x = self.regressive_head(x[:, -1, :])
+        return x, (hidden_state, cell_state)

{deeplotx-0.3.1 → deeplotx-0.4.1}/deeplotx.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.3.1
+Version: 0.4.1
 Summary: Easy-2-use long text classifier trainers.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: huggingface-hub[hf-xet]>=0.30.2
+Requires-Dist: hf-xet>=1.0.5
 Requires-Dist: jupyter>=1.1.1
 Requires-Dist: numpy>=2.2.5
 Requires-Dist: python-dotenv>=1.1.0

{deeplotx-0.3.1 → deeplotx-0.4.1}/deeplotx.egg-info/SOURCES.txt

@@ -12,9 +12,11 @@ deeplotx/encoder/bert_encoder.py
 deeplotx/encoder/long_text_encoder.py
 deeplotx/encoder/longformer_encoder.py
 deeplotx/nn/__init__.py
+deeplotx/nn/auto_regression.py
 deeplotx/nn/base_neural_network.py
 deeplotx/nn/linear_regression.py
 deeplotx/nn/logistic_regression.py
+deeplotx/nn/recursive_sequential.py
 deeplotx/nn/softmax_regression.py
 deeplotx/similarity/__init__.py
 deeplotx/similarity/distribution.py

{deeplotx-0.3.1 → deeplotx-0.4.1}/deeplotx.egg-info/requires.txt

@@ -1,4 +1,4 @@
-huggingface-hub[hf-xet]>=0.30.2
+hf-xet>=1.0.5
 jupyter>=1.1.1
 numpy>=2.2.5
 python-dotenv>=1.1.0

{deeplotx-0.3.1 → deeplotx-0.4.1}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "deeplotx"
-version = "0.3.1"
+version = "0.4.1"
 description = "Easy-2-use long text classifier trainers."
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "huggingface-hub[hf-xet]>=0.30.2",
+    "hf-xet>=1.0.5",
     "jupyter>=1.1.1",
     "numpy>=2.2.5",
     "python-dotenv>=1.1.0",