dragon-ml-toolbox 5.2.1__py3-none-any.whl → 5.2.2__py3-none-any.whl

{dragon_ml_toolbox-5.2.1.dist-info → dragon_ml_toolbox-5.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 5.2.1
+Version: 5.2.2
 Summary: A collection of tools for data science and machine learning projects.
 Author-email: Karl Loza <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-5.2.1.dist-info → dragon_ml_toolbox-5.2.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-dragon_ml_toolbox-5.2.1.dist-info/licenses/LICENSE,sha256=2uUFNy7D0TLgHim1K5s3DIJ4q_KvxEXVilnU20cWliY,1066
-dragon_ml_toolbox-5.2.1.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=lY4_rJPnLnMu7YBQaY-_iz1JRDcLdQzNCyeLAF1glJY,1837
+dragon_ml_toolbox-5.2.2.dist-info/licenses/LICENSE,sha256=2uUFNy7D0TLgHim1K5s3DIJ4q_KvxEXVilnU20cWliY,1066
+dragon_ml_toolbox-5.2.2.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=lY4_rJPnLnMu7YBQaY-_iz1JRDcLdQzNCyeLAF1glJY,1837
 ml_tools/ETL_engineering.py,sha256=4wwZXi9_U7xfCY70jGBaKniOeZ0m75ppxWpQBd_DmLc,39369
 ml_tools/GUI_tools.py,sha256=n4ZZ5kEjwK5rkOCFJE41HeLFfjhpJVLUSzk9Kd9Kr_0,45410
 ml_tools/MICE_imputation.py,sha256=b6ZTs8RedXFifOpuMCzr68xM16mCBVh1Ua6kcGfiVtg,11462
@@ -7,7 +7,7 @@ ml_tools/ML_callbacks.py,sha256=xiJ6NnoVwF_TVak6sYzwWFk4CI3vRJGjxvGI1Yq6euw,1333
 ml_tools/ML_datasetmaster.py,sha256=IzT2v1o71PgYCFi9RXccBnmH-t-ExzX8sn9cCD2gz-Y,33603
 ml_tools/ML_evaluation.py,sha256=4dVqe6JF1Ukmk1sAcY8E5EG1oB1_oy2HXE5OT-pZwCs,10273
 ml_tools/ML_inference.py,sha256=Fh-X2UQn3AznWBjf-7iPSxwE-EzkGQm1VEIRUAkURmE,5336
-ml_tools/ML_models.py,sha256=vG4i_lmQ9Jz6twO2jmu6teHET8qluKO02odSwzYfvRI,5436
+ml_tools/ML_models.py,sha256=SJhKHGAN2VTBqzcHUOpFWuVZ2Y7U1M4P_axG_LNYWcI,6460
 ml_tools/ML_optimization.py,sha256=2L9BSUzgLOEwBU84TN1qDh1KAOJ4R6C6NYSe7jmE4RI,9656
 ml_tools/ML_trainer.py,sha256=t58Ka6ryaYm0Fi5xje-e-fkmz9DwDLIeJLbh04n_gDg,15034
 ml_tools/PSO_optimization.py,sha256=stH2Ux1sftQgX5EwLc85kHcoT4Rmz6zv7sH2yzf4Zrw,22710
@@ -26,7 +26,7 @@ ml_tools/keys.py,sha256=kK9UF-hek2VcPGFILCKl5geoN6flmMOu7IzhdEA6z5Y,1068
 ml_tools/optimization_tools.py,sha256=MuT4OG7_r1QqLUti-yYix7QeCpglezD0oe9BDCq0QXk,5086
 ml_tools/path_manager.py,sha256=Z8e7w3MPqQaN8xmTnKuXZS6CIW59BFwwqGhGc00sdp4,13692
 ml_tools/utilities.py,sha256=mz-M351DzxWxnYVcLX-7ZQ6c-RGoCV9g4VTS9Qif2Es,18348
-dragon_ml_toolbox-5.2.1.dist-info/METADATA,sha256=aOUNeJp2N8iT0y85jlp7N_6PsRbOnrBJgS5FhiVCjWs,6638
-dragon_ml_toolbox-5.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dragon_ml_toolbox-5.2.1.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
-dragon_ml_toolbox-5.2.1.dist-info/RECORD,,
+dragon_ml_toolbox-5.2.2.dist-info/METADATA,sha256=1xc1_iWoGsLxwEFcyLRRSJCJJNdQZNsVHCSykfaVKGQ,6638
+dragon_ml_toolbox-5.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-5.2.2.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-5.2.2.dist-info/RECORD,,

ml_tools/ML_models.py

@@ -66,6 +66,21 @@ class MultilayerPerceptron(nn.Module):
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         """Defines the forward pass of the model."""
         return self._layers(x)
+    
+    def __repr__(self) -> str:
+        """Returns the developer-friendly string representation of the model."""
+        # Extracts the number of neurons from each nn.Linear layer
+        layer_sizes = [layer.in_features for layer in self._layers if isinstance(layer, nn.Linear)]
+        
+        # Get the last layer and check its type before accessing the attribute
+        last_layer = self._layers[-1]
+        if isinstance(last_layer, nn.Linear):
+            layer_sizes.append(last_layer.out_features)
+        
+        # Creates a string like: 10 -> 40 -> 80 -> 40 -> 2
+        arch_str = ' -> '.join(map(str, layer_sizes))
+        
+        return f"MultilayerPerceptron(arch: {arch_str})"
 
 
 class SequencePredictorLSTM(nn.Module):
@@ -128,6 +143,14 @@ class SequencePredictorLSTM(nn.Module):
         predictions = self.linear(lstm_out)
         
         return predictions
+    
+    def __repr__(self) -> str:
+        """Returns the developer-friendly string representation of the model."""
+        return (
+            f"SequencePredictorLSTM(features={self.lstm.input_size}, "
+            f"hidden_size={self.lstm.hidden_size}, "
+            f"recurrent_layers={self.lstm.num_layers})"
+        )
 
 
 def info():