deeplotx 0.2.19__tar.gz → 0.2.20__tar.gz

{deeplotx-0.2.19 → deeplotx-0.2.20}/PKG-INFO

@@ -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 → deeplotx-0.2.20}/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 → deeplotx-0.2.20}/deeplotx.egg-info/PKG-INFO

@@ -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 → deeplotx-0.2.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "deeplotx"
-version = "0.2.19"
+version = "0.2.20"
 description = "Easy-2-use long text classifier trainers."
 readme = "README.md"
 requires-python = ">=3.10"