deeplotx 0.2.19__py3-none-any.whl → 0.2.20__py3-none-any.whl

deeplotx/nn/base_neural_network.py

@@ -23,10 +23,10 @@ class BaseNeuralNetwork(nn.Module):
             for param in self.parameters():
                 l2_reg += (torch.pow(param, exponent=2.)).sum()
             return l2_reg
-        return _lambda * _l2()
+        return _lambda * _l2() / 2.
 
     def elastic_net(self, alpha: float = 1e-4, rho: float = 0.5) -> torch.Tensor:
-        return alpha * rho * self.l1(_lambda=1.) + alpha * (1 - rho) * self.l2(_lambda=1.) / 2.
+        return alpha * (rho * self.l1(_lambda=1.) + (1 - rho) * self.l2(_lambda=1.))
 
     def forward(self, x) -> torch.Tensor: ...
 
@@ -39,9 +39,9 @@ class BaseNeuralNetwork(nn.Module):
         return res
 
     def save(self):
-        torch.save(self.state_dict(), f'{self._model_name}.deeplotx.pth')
+        torch.save(self.state_dict(), f'{self._model_name}.deeplotx')
         return self
 
     def load(self):
-        self.load_state_dict(torch.load(f'{self._model_name}.deeplotx.pth'))
+        self.load_state_dict(torch.load(f'{self._model_name}.deeplotx'))
         return self

{deeplotx-0.2.19.dist-info → deeplotx-0.2.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.2.19
+Version: 0.2.20
 Summary: Easy-2-use long text classifier trainers.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{deeplotx-0.2.19.dist-info → deeplotx-0.2.20.dist-info}/RECORD

@@ -3,7 +3,7 @@ deeplotx/encoder/__init__.py,sha256=x7k8IE0FXvDl7kCJGWPsetOHFdvNCiCXHbYOdvo7_JQ,
 deeplotx/encoder/bert_encoder.py,sha256=rdT8YgZzvRoqYqtzPW95ilagSQTAQgUl7mMVetGKxCY,1822
 deeplotx/encoder/long_text_encoder.py,sha256=yEEtTVZYHJ0W3OSbh7BHm6xI33nJmVYlSrgD5RVcJLY,2967
 deeplotx/nn/__init__.py,sha256=9gh8rhKqVWtJyvryU_wHPTLEQIorwOBhAQRc0DtNamM,153
-deeplotx/nn/base_neural_network.py,sha256=EjBqW_8DMNX9TqNzOaLsLi5RzgsCyBaCgwuVque4jwk,1580
+deeplotx/nn/base_neural_network.py,sha256=MXuID5bagdHyrFOkoybW1oiXAY2d4FGnzZoR37LZfUI,1566
 deeplotx/nn/linear_regression.py,sha256=D4mEWVOq6q1Fm2otm57rgZ_E06HJLZBV5k636PprAf4,1520
 deeplotx/nn/logistic_regression.py,sha256=QAtZp2oyqOW8-1pJWVcahsSM83bzfA68EHObg-wSHHY,463
 deeplotx/nn/softmax_regression.py,sha256=eUn3mVNlye9ewVdw3McPHZuKbUvvaamsUgFIJMVMgBU,487
@@ -13,8 +13,8 @@ deeplotx/trainer/text_binary_classification_trainer.py,sha256=5O-5dwVMCj5EDX9gjJ
 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.2.19.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
-deeplotx-0.2.19.dist-info/METADATA,sha256=SLrWD0GXTXJW4I1_Wv7JXwfjJOR0e2Nz-CBDg7yd_-Y,1573
-deeplotx-0.2.19.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-deeplotx-0.2.19.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
-deeplotx-0.2.19.dist-info/RECORD,,
+deeplotx-0.2.20.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
+deeplotx-0.2.20.dist-info/METADATA,sha256=NQgRWucDSAI4awAJNf9984IujFRo9PurR1qrqpmWIzA,1573
+deeplotx-0.2.20.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+deeplotx-0.2.20.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
+deeplotx-0.2.20.dist-info/RECORD,,