deeplotx 0.3.2__tar.gz → 0.4.2__tar.gz

{deeplotx-0.3.2 → deeplotx-0.4.2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.3.2
-Summary: Easy-2-use long text classifier trainers.
+Version: 0.4.2
+Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -14,7 +14,7 @@ Requires-Dist: transformers>=4.51.3
 Requires-Dist: typing-extensions>=4.13.2
 Dynamic: license-file
 
-# DeepLoTX: Easy2UseLongTextClassifierTrainers
+# DeepLoTX: Easy2UseLongTextNLPToolkit
 
 ## Installation
 
@@ -38,7 +38,7 @@ Dynamic: license-file
 
 ## Quick Start
 
-To train a binary classifier for text files:
+To train a binary classifier from text files:
 
 ```python
 from deeplotx.util import get_files, read_file

{deeplotx-0.3.2 → deeplotx-0.4.2}/README.md

@@ -1,4 +1,4 @@
-# DeepLoTX: Easy2UseLongTextClassifierTrainers
+# DeepLoTX: Easy2UseLongTextNLPToolkit
 
 ## Installation
 
@@ -22,7 +22,7 @@
 
 ## Quick Start
 
-To train a binary classifier for text files:
+To train a binary classifier from text files:
 
 ```python
 from deeplotx.util import get_files, read_file

{deeplotx-0.3.2 → deeplotx-0.4.2}/deeplotx/__init__.py

@@ -4,7 +4,13 @@ import os
 __ROOT__ = os.path.dirname(os.path.abspath(__file__))
 
 from .encoder import BertEncoder, LongTextEncoder, LongformerEncoder
-from .nn import LinearRegression, LogisticRegression, SoftmaxRegression
+from .nn import (
+    LinearRegression,
+    LogisticRegression,
+    SoftmaxRegression,
+    RecursiveSequential,
+    AutoRegression
+)
 from .trainer import TextBinaryClassifierTrainer
 
 __AUTHOR__ = '吴子豪 / Vortez Wohl'

{deeplotx-0.3.2 → deeplotx-0.4.2}/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.2/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.2 → deeplotx-0.4.2}/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.2/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.2 → deeplotx-0.4.2}/deeplotx.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.3.2
-Summary: Easy-2-use long text classifier trainers.
+Version: 0.4.2
+Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -14,7 +14,7 @@ Requires-Dist: transformers>=4.51.3
 Requires-Dist: typing-extensions>=4.13.2
 Dynamic: license-file
 
-# DeepLoTX: Easy2UseLongTextClassifierTrainers
+# DeepLoTX: Easy2UseLongTextNLPToolkit
 
 ## Installation
 
@@ -38,7 +38,7 @@ Dynamic: license-file
 
 ## Quick Start
 
-To train a binary classifier for text files:
+To train a binary classifier from text files:
 
 ```python
 from deeplotx.util import get_files, read_file

{deeplotx-0.3.2 → deeplotx-0.4.2}/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.2 → deeplotx-0.4.2}/pyproject.toml

@@ -1,15 +1,15 @@
-[project]
-name = "deeplotx"
-version = "0.3.2"
-description = "Easy-2-use long text classifier trainers."
-readme = "README.md"
-requires-python = ">=3.10"
-dependencies = [
-    "hf-xet>=1.0.5",
-    "jupyter>=1.1.1",
-    "numpy>=2.2.5",
-    "python-dotenv>=1.1.0",
-    "torch>=2.7.0",
-    "transformers>=4.51.3",
-    "typing-extensions>=4.13.2",
-]
+[project]
+name = "deeplotx"
+version = "0.4.2"
+description = "Easy-2-use long text NLP toolkit."
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "hf-xet>=1.0.5",
+    "jupyter>=1.1.1",
+    "numpy>=2.2.5",
+    "python-dotenv>=1.1.0",
+    "torch>=2.7.0",
+    "transformers>=4.51.3",
+    "typing-extensions>=4.13.2",
+]