deeplotx 0.2.17__tar.gz → 0.2.19__tar.gz

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

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

{deeplotx-0.2.17 → deeplotx-0.2.19}/deeplotx/nn/base_neural_network.py

@@ -25,9 +25,8 @@ class BaseNeuralNetwork(nn.Module):
             return l2_reg
         return _lambda * _l2()
 
-    def elastic_net(self, alpha: float = 1e-4, rho: float = 0.5,
-                    lambda_l1: float = 1e-4, lambda_l2: float = 1e-4) -> torch.Tensor:
-        return alpha * rho * self.l1(_lambda=lambda_l1) + alpha * (1 - rho) * self.l2(_lambda=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.
 
     def forward(self, x) -> torch.Tensor: ...
 

{deeplotx-0.2.17 → deeplotx-0.2.19}/deeplotx/trainer/text_binary_classification_trainer.py

@@ -21,8 +21,7 @@ class TextBinaryClassifierTrainer(BaseTrainer):
     def train(self, positive_texts: list[str], negative_texts: list[str],
               num_epochs: int, learning_rate: float = 2e-5, balancing_dataset: bool = True,
               train_loss_threshold: float = 0.0, valid_loss_threshold: float = 0.0,
-              alpha: float = 1e-4, rho: float = 0.2,
-              lambda_l1: float = 1e-4, lambda_l2: float = 1e-4) -> LogisticRegression:
+              alpha: float = 1e-4, rho: float = 0.2) -> LogisticRegression:
         if balancing_dataset:
             min_length = min(len(positive_texts), len(negative_texts))
             positive_texts = positive_texts[:min_length]
@@ -52,10 +51,7 @@ class TextBinaryClassifierTrainer(BaseTrainer):
             total_loss = 0.0
             for batch_texts, batch_labels in train_loader:
                 outputs = self.model.forward(batch_texts)
-                loss = loss_function(outputs, batch_labels) + self.model.elastic_net(
-                    alpha=alpha, rho=rho,
-                    lambda_l1=lambda_l1, lambda_l2=lambda_l2
-                )
+                loss = loss_function(outputs, batch_labels) + self.model.elastic_net(alpha=alpha, rho=rho)
                 optimizer.zero_grad()
                 loss.backward()
                 optimizer.step()
@@ -66,10 +62,7 @@ class TextBinaryClassifierTrainer(BaseTrainer):
                     with torch.no_grad():
                         self.model.eval()
                         outputs = self.model.forward(batch_texts)
-                        loss = loss_function(outputs, batch_labels) + self.model.elastic_net(
-                            alpha=alpha, rho=rho,
-                            lambda_l1=lambda_l1, lambda_l2=lambda_l2
-                        )
+                        loss = loss_function(outputs, batch_labels) + self.model.elastic_net(alpha=alpha, rho=rho)
                         total_valid_loss += loss.item()
                         self.model.train()
                 logger.debug(f"Epoch {epoch + 1}/{num_epochs} | "

{deeplotx-0.2.17 → deeplotx-0.2.19}/deeplotx.egg-info/PKG-INFO

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

{deeplotx-0.2.17 → deeplotx-0.2.19}/pyproject.toml

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