deeplotx 0.4.12b3__py3-none-any.whl → 0.4.12b5__py3-none-any.whl

deeplotx/nn/base_neural_network.py

@@ -24,7 +24,7 @@ class BaseNeuralNetwork(nn.Module):
 
     def l1(self, _lambda: float = 1e-4) -> torch.Tensor:
         def _l1() -> torch.Tensor:
-            l2_reg = torch.tensor(0.)
+            l2_reg = torch.tensor(0., device=self.device, dtype=self.dtype)
             for param in self.parameters():
                 l2_reg += (torch.abs(param)).sum()
             return l2_reg
@@ -32,7 +32,7 @@ class BaseNeuralNetwork(nn.Module):
 
     def l2(self, _lambda: float = 1e-4) -> torch.Tensor:
         def _l2() -> torch.Tensor:
-            l2_reg = torch.tensor(0.)
+            l2_reg = torch.tensor(0., device=self.device, dtype=self.dtype)
             for param in self.parameters():
                 l2_reg += (torch.pow(param, exponent=2.)).sum()
             return l2_reg
@@ -45,6 +45,7 @@ class BaseNeuralNetwork(nn.Module):
     def forward(self, *args, **kwargs) -> torch.Tensor: ...
 
     def predict(self, x) -> torch.Tensor:
+        x = self.ensure_device_and_dtype(x, device=self.device, dtype=self.dtype)
         __train = self.training
         self.training = False
         with torch.no_grad():

deeplotx/nn/recursive_sequential.py

@@ -34,3 +34,13 @@ class RecursiveSequential(BaseNeuralNetwork):
         x, (hidden_state, cell_state) = self.lstm(x, state)
         x = self.regressive_head(x[:, -1, :])
         return x, (hidden_state, cell_state)
+
+    @override
+    def predict(self, x, batch_size: int | None = None) -> torch.Tensor:
+        _batch_size = batch_size if batch_size is not None else x.shape[0]
+        __train = self.training
+        self.training = False
+        with torch.no_grad():
+            res = self.forward(x, _batch_size)[0]
+        self.training = __train
+        return res

{deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.4.12b3
+Version: 0.4.12b5
 Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b5.dist-info}/RECORD

@@ -5,10 +5,10 @@ deeplotx/encoder/long_text_encoder.py,sha256=hl_O8kR9o1kcII9YfSx2rf_Pk0l_Rv7LNbs
 deeplotx/encoder/longformer_encoder.py,sha256=A8FXqd4mdHxSn_o_R689XtpT73ISDT788EgMQRGLC2g,1822
 deeplotx/nn/__init__.py,sha256=oQ-vYXyuaGelfCOs2im_gZXAiiBlCCVXh1uw9yjvRMs,253
 deeplotx/nn/auto_regression.py,sha256=7P63opWCWMqE2DigwbsL6kfXtFtJPz00Yo1RqflBz4A,572
-deeplotx/nn/base_neural_network.py,sha256=yEyF5C-Z3bp4Ddx6GbvqpBxXyFdbSChmP6SgyTzjQmM,2180
+deeplotx/nn/base_neural_network.py,sha256=ufA0QOFFXaz4RLqjqx9N6VY-mDwWOe9Y35u2vsh_NFc,2339
 deeplotx/nn/linear_regression.py,sha256=_LQFrOKBbQxvuNzb_B8Mr6PAQJUg-pFeu3h7_jQz04o,2166
 deeplotx/nn/logistic_regression.py,sha256=j8QGe0e7In97RMOXApJRID85qf1rOUCOk3V368CBfqs,653
-deeplotx/nn/recursive_sequential.py,sha256=YCQUUcTBsZUeyO7CLjUO1EISYX1SXPnW6asR6ZBQAb4,1926
+deeplotx/nn/recursive_sequential.py,sha256=4NeSE11ZsZ4YBpjee2mRn3a3hqU4xw_k6t09zTl-tTg,2292
 deeplotx/nn/softmax_regression.py,sha256=SlhvHho-Oufp7adAjm1t1ygidu-FrnHQ9aleMXyS_s8,674
 deeplotx/similarity/__init__.py,sha256=s3u-KSgxjnMcWpIItKgXNltFMPQ7YY3CqsqHI-5F1c8,724
 deeplotx/similarity/distribution.py,sha256=wQGouuuW531pZeBRKBujXsdsoz4fDnPw7_GW81jwepc,1066
@@ -20,8 +20,8 @@ deeplotx/trainer/text_binary_classification_trainer.py,sha256=Wq_pGO78zgdXxFeBja
 deeplotx/util/__init__.py,sha256=JxqAK_WOOHcYVSTHBT1-WuBwWrPEVDTV3titeVWvNUM,74
 deeplotx/util/hash.py,sha256=wwsC6kOQvbpuvwKsNQOARd78_wePmW9i3oaUuXRUnpc,352
 deeplotx/util/read_file.py,sha256=ptzouvEQeeW8KU5BrWNJlXw-vFXVrpS9SkAUxsu6A8A,612
-deeplotx-0.4.12b3.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
-deeplotx-0.4.12b3.dist-info/METADATA,sha256=HZY8s697peX3onvVr5bjdHqwKJk2YyMOadMLGk-QvSc,6287
-deeplotx-0.4.12b3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deeplotx-0.4.12b3.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
-deeplotx-0.4.12b3.dist-info/RECORD,,
+deeplotx-0.4.12b5.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
+deeplotx-0.4.12b5.dist-info/METADATA,sha256=xA4HtA5PMPO5goVZITcOJs6hWrFD1DNXDZASs5dcD_o,6287
+deeplotx-0.4.12b5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deeplotx-0.4.12b5.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
+deeplotx-0.4.12b5.dist-info/RECORD,,