deeplotx 0.4.12b3__py3-none-any.whl → 0.4.12b4__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- deeplotx/nn/base_neural_network.py +3 -2
- {deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b4.dist-info}/METADATA +1 -1
- {deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b4.dist-info}/RECORD +6 -6
- {deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b4.dist-info}/WHEEL +0 -0
- {deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b4.dist-info}/licenses/LICENSE +0 -0
- {deeplotx-0.4.12b3.dist-info → deeplotx-0.4.12b4.dist-info}/top_level.txt +0 -0
|
@@ -24,7 +24,7 @@ class BaseNeuralNetwork(nn.Module):
|
|
|
24
24
|
|
|
25
25
|
def l1(self, _lambda: float = 1e-4) -> torch.Tensor:
|
|
26
26
|
def _l1() -> torch.Tensor:
|
|
27
|
-
l2_reg = torch.tensor(0.)
|
|
27
|
+
l2_reg = torch.tensor(0., device=self.device, dtype=self.dtype)
|
|
28
28
|
for param in self.parameters():
|
|
29
29
|
l2_reg += (torch.abs(param)).sum()
|
|
30
30
|
return l2_reg
|
|
@@ -32,7 +32,7 @@ class BaseNeuralNetwork(nn.Module):
|
|
|
32
32
|
|
|
33
33
|
def l2(self, _lambda: float = 1e-4) -> torch.Tensor:
|
|
34
34
|
def _l2() -> torch.Tensor:
|
|
35
|
-
l2_reg = torch.tensor(0.)
|
|
35
|
+
l2_reg = torch.tensor(0., device=self.device, dtype=self.dtype)
|
|
36
36
|
for param in self.parameters():
|
|
37
37
|
l2_reg += (torch.pow(param, exponent=2.)).sum()
|
|
38
38
|
return l2_reg
|
|
@@ -45,6 +45,7 @@ class BaseNeuralNetwork(nn.Module):
|
|
|
45
45
|
def forward(self, *args, **kwargs) -> torch.Tensor: ...
|
|
46
46
|
|
|
47
47
|
def predict(self, x) -> torch.Tensor:
|
|
48
|
+
x = self.ensure_device_and_dtype(x, device=self.device, dtype=self.dtype)
|
|
48
49
|
__train = self.training
|
|
49
50
|
self.training = False
|
|
50
51
|
with torch.no_grad():
|
|
@@ -5,7 +5,7 @@ deeplotx/encoder/long_text_encoder.py,sha256=hl_O8kR9o1kcII9YfSx2rf_Pk0l_Rv7LNbs
|
|
|
5
5
|
deeplotx/encoder/longformer_encoder.py,sha256=A8FXqd4mdHxSn_o_R689XtpT73ISDT788EgMQRGLC2g,1822
|
|
6
6
|
deeplotx/nn/__init__.py,sha256=oQ-vYXyuaGelfCOs2im_gZXAiiBlCCVXh1uw9yjvRMs,253
|
|
7
7
|
deeplotx/nn/auto_regression.py,sha256=7P63opWCWMqE2DigwbsL6kfXtFtJPz00Yo1RqflBz4A,572
|
|
8
|
-
deeplotx/nn/base_neural_network.py,sha256=
|
|
8
|
+
deeplotx/nn/base_neural_network.py,sha256=ufA0QOFFXaz4RLqjqx9N6VY-mDwWOe9Y35u2vsh_NFc,2339
|
|
9
9
|
deeplotx/nn/linear_regression.py,sha256=_LQFrOKBbQxvuNzb_B8Mr6PAQJUg-pFeu3h7_jQz04o,2166
|
|
10
10
|
deeplotx/nn/logistic_regression.py,sha256=j8QGe0e7In97RMOXApJRID85qf1rOUCOk3V368CBfqs,653
|
|
11
11
|
deeplotx/nn/recursive_sequential.py,sha256=YCQUUcTBsZUeyO7CLjUO1EISYX1SXPnW6asR6ZBQAb4,1926
|
|
@@ -20,8 +20,8 @@ deeplotx/trainer/text_binary_classification_trainer.py,sha256=Wq_pGO78zgdXxFeBja
|
|
|
20
20
|
deeplotx/util/__init__.py,sha256=JxqAK_WOOHcYVSTHBT1-WuBwWrPEVDTV3titeVWvNUM,74
|
|
21
21
|
deeplotx/util/hash.py,sha256=wwsC6kOQvbpuvwKsNQOARd78_wePmW9i3oaUuXRUnpc,352
|
|
22
22
|
deeplotx/util/read_file.py,sha256=ptzouvEQeeW8KU5BrWNJlXw-vFXVrpS9SkAUxsu6A8A,612
|
|
23
|
-
deeplotx-0.4.
|
|
24
|
-
deeplotx-0.4.
|
|
25
|
-
deeplotx-0.4.
|
|
26
|
-
deeplotx-0.4.
|
|
27
|
-
deeplotx-0.4.
|
|
23
|
+
deeplotx-0.4.12b4.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
|
|
24
|
+
deeplotx-0.4.12b4.dist-info/METADATA,sha256=v6pgBom07_sEXSaipdfGnk1hueBE_1zjZ323xPlwglc,6287
|
|
25
|
+
deeplotx-0.4.12b4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
26
|
+
deeplotx-0.4.12b4.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
|
|
27
|
+
deeplotx-0.4.12b4.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|