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

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.18.dist-info → deeplotx-0.2.19.dist-info}/METADATA

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

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

@@ -9,12 +9,12 @@ deeplotx/nn/logistic_regression.py,sha256=QAtZp2oyqOW8-1pJWVcahsSM83bzfA68EHObg-
 deeplotx/nn/softmax_regression.py,sha256=eUn3mVNlye9ewVdw3McPHZuKbUvvaamsUgFIJMVMgBU,487
 deeplotx/trainer/__init__.py,sha256=Fl5DR9UecQc5VtBcczU9sx_HtPNoFohpuELOh-Jrsks,77
 deeplotx/trainer/base_trainer.py,sha256=z0MeAT-rRYmjeBXt0ckt7J1itYArR0Cx02wHesXUoZE,385
-deeplotx/trainer/text_binary_classification_trainer.py,sha256=SEisKhDhcOoV87N0xLLVwuZrDSadcHmdoFIji-RqjT4,4375
+deeplotx/trainer/text_binary_classification_trainer.py,sha256=5O-5dwVMCj5EDX9gjJwCA468OR4UozJ7V8b-JxeUB0s,4080
 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.18.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
-deeplotx-0.2.18.dist-info/METADATA,sha256=fn9tKJrvY8_JnnnpMG4N6EajaTI3HgN2iRrFodc5SE0,1573
-deeplotx-0.2.18.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-deeplotx-0.2.18.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
-deeplotx-0.2.18.dist-info/RECORD,,
+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,,