deeplotx 0.4.13__py3-none-any.whl → 0.4.15__py3-none-any.whl

deeplotx/nn/base_neural_network.py

@@ -58,5 +58,5 @@ class BaseNeuralNetwork(nn.Module):
         return self
 
     def load(self):
-        self.load_state_dict(torch.load(f'{self._model_name}.deeplotx'))
+        self.load_state_dict(torch.load(f'{self._model_name}.deeplotx', map_location=self.device, weights_only=True))
         return self

{deeplotx-0.4.13.dist-info → deeplotx-0.4.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.4.13
+Version: 0.4.15
 Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -177,31 +177,33 @@ Dynamic: license-file
 
     import torch
     from torch import nn
-
+    
     from deeplotx.nn.base_neural_network import BaseNeuralNetwork
-
-
+    
+    
     class LinearRegression(BaseNeuralNetwork):
-        def __init__(self, input_dim: int, output_dim: int, model_name: str | None = None):
-            super().__init__(model_name=model_name)
-            self.fc1 = nn.Linear(input_dim, 1024)
-            self.fc1_to_fc4_res = nn.Linear(1024, 64)
-            self.fc2 = nn.Linear(1024, 768)
-            self.fc3 = nn.Linear(768, 128)
-            self.fc4 = nn.Linear(128, 64)
-            self.fc5 = nn.Linear(64, output_dim)
-            self.parametric_relu_1 = nn.PReLU(num_parameters=1, init=5e-3)
-            self.parametric_relu_2 = nn.PReLU(num_parameters=1, init=5e-3)
-            self.parametric_relu_3 = nn.PReLU(num_parameters=1, init=5e-3)
-            self.parametric_relu_4 = nn.PReLU(num_parameters=1, init=5e-3)
-
+        def __init__(self, input_dim: int, output_dim: int, model_name: str | None = None,
+                     device: str | None = None, dtype: torch.dtype | None = None):
+            super().__init__(model_name=model_name, device=device, dtype=dtype)
+            self.fc1 = nn.Linear(input_dim, 1024, device=self.device, dtype=self.dtype)
+            self.fc1_to_fc4_res = nn.Linear(1024, 64, device=self.device, dtype=self.dtype)
+            self.fc2 = nn.Linear(1024, 768, device=self.device, dtype=self.dtype)
+            self.fc3 = nn.Linear(768, 128, device=self.device, dtype=self.dtype)
+            self.fc4 = nn.Linear(128, 64, device=self.device, dtype=self.dtype)
+            self.fc5 = nn.Linear(64, output_dim, device=self.device, dtype=self.dtype)
+            self.parametric_relu_1 = nn.PReLU(num_parameters=1, init=5e-3, device=self.device, dtype=self.dtype)
+            self.parametric_relu_2 = nn.PReLU(num_parameters=1, init=5e-3, device=self.device, dtype=self.dtype)
+            self.parametric_relu_3 = nn.PReLU(num_parameters=1, init=5e-3, device=self.device, dtype=self.dtype)
+            self.parametric_relu_4 = nn.PReLU(num_parameters=1, init=5e-3, device=self.device, dtype=self.dtype)
+    
         @override
         def forward(self, x) -> torch.Tensor:
+            x = self.ensure_device_and_dtype(x, device=self.device, dtype=self.dtype)
             fc1_out = self.parametric_relu_1(self.fc1(x))
-            x = nn.LayerNorm(normalized_shape=1024, eps=1e-9)(fc1_out)
+            x = nn.LayerNorm(normalized_shape=1024, eps=1e-9, device=self.device, dtype=self.dtype)(fc1_out)
             x = torch.dropout(x, p=0.2, train=self.training)
             x = self.parametric_relu_2(self.fc2(x))
-            x = nn.LayerNorm(normalized_shape=768, eps=1e-9)(x)
+            x = nn.LayerNorm(normalized_shape=768, eps=1e-9, device=self.device, dtype=self.dtype)(x)
             x = torch.dropout(x, p=0.2, train=self.training)
             x = self.parametric_relu_3(self.fc3(x))
             x = torch.dropout(x, p=0.2, train=self.training)

{deeplotx-0.4.13.dist-info → deeplotx-0.4.15.dist-info}/RECORD

@@ -5,7 +5,7 @@ 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=ufA0QOFFXaz4RLqjqx9N6VY-mDwWOe9Y35u2vsh_NFc,2339
+deeplotx/nn/base_neural_network.py,sha256=oGlqY6ZZ3DGOXWQQ9nZ7ktJpfyIaUrSraGnFRbGD1jM,2384
 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=pHZChjzw9cuMQ0lmv42lxxVgxCU6D7owOgph0Irj-w4,2219
@@ -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.13.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
-deeplotx-0.4.13.dist-info/METADATA,sha256=IpaDS2hmLGYUDKEhE_o-uTR0Go2ZkAFQCHQMsft2-o8,6285
-deeplotx-0.4.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-deeplotx-0.4.13.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
-deeplotx-0.4.13.dist-info/RECORD,,
+deeplotx-0.4.15.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
+deeplotx-0.4.15.dist-info/METADATA,sha256=HB6VHdLgyuMclJYLykBMnbnLa7s-rwfHyhrgjNdoRFQ,6955
+deeplotx-0.4.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+deeplotx-0.4.15.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
+deeplotx-0.4.15.dist-info/RECORD,,