NeuralNetworks 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

NeuralNetworks/Dependances/pytorch.py

@@ -6,7 +6,6 @@
 # (at your option) any later version.
 
 import os
-os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 
 import platform
 
@@ -20,6 +19,8 @@ from torch.utils.data import TensorDataset, DataLoader
 from torchmetrics.image import PeakSignalNoiseRatio as PSNR
 from torchvision.transforms import ToTensor, Resize, Compose
 
+import visualtorch
+
 torch.cuda.empty_cache()
 def get_best_device():
 
@@ -51,6 +52,7 @@ def get_best_device():
 
     # =========== Unknown OS ===========
     return torch.device("cpu")
+    
 device = get_best_device()
 
 # --- Optimisations CUDA ---

NeuralNetworks/MLP/__init__.py

@@ -10,7 +10,9 @@ from .FourierFeatures import encode
 from .Layers import create_layers
 from .inference import infer
 
-class MLP (nn.Module):
+from ..shared import Module
+
+class MLP (Module):
     """
     Multi-Layer Perceptron avec encodage de Fourier optionnel.
 
@@ -49,18 +51,13 @@ class MLP (nn.Module):
         nb_fourier = 8,
         norm = "Relu",
         name = "Net"):
-        super ().__init__ ()
+        super ().__init__ (name)
 
         # --- Activation ---
         self.norm = norm_list.get (norm)
         if self.norm is None:
-            print (f"Warning: '{norm}' not recognized, falling back to 'Relu'")
+            print (f"Warning: '{norm}' not recognized, fRelung back to 'm is'")
             self.norm = norm_list.get ("Relu")
-
-        # --- Attributs ---
-        self.losses = []
-        self.learnings = []
-        self.name = name
         
         ## --- Encodage Fourier ou passthrough ---
         self.encodings, self.f = encode (
@@ -82,18 +79,8 @@ class MLP (nn.Module):
             self.norm
         )
 
-    def forward (self, x):
-        """
-        Effectue une passe avant du réseau.
-
-        Parameters
-        ----------
-        x : torch.Tensor
-            Entrées du réseau de shape `(N, input_size)`.
+    def _forward (self, x):
+        return infer (self, x)
 
-        Returns
-        -------
-        torch.Tensor
-            Sortie du MLP de shape `(N, output_size)`.
-        """
+    def train_forward (self, x):
         return infer (self, x)

NeuralNetworks/MLP/inference.py

@@ -8,19 +8,5 @@
 from ..Dependances import torch, np, device
 
 def infer (net, x):
-    with torch.no_grad ():
-        x = x.unsqueeze (0) if x.dim () == 1 else x
-
-        net = net.to (device)
-        x = x.to (device)
-        results_list = [net.model (encoding (x)) for encoding in net.encodings]
-        x = x.to ('cpu')
-
-        output = np.array (
-            net.f (
-                torch.cat (results_list, dim = 1)
-            ).cpu ().numpy ().flatten ()
-        )
-        net = net.to ('cpu')
-
-    return output
+    results_list = [net.model (encoding (x)) for encoding in net.encodings]
+    return net.f (torch.cat (results_list, dim = 1))

NeuralNetworks/Trainer/__init__.py

@@ -5,7 +5,7 @@
 # the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 
-from ..Dependances import crit_list, optim_list
+from ..Dependances import crit_list, optim_list, np
 from .train import train_f
 from .sample_data import sample_data
 
@@ -15,7 +15,7 @@ class Trainer:
         *nets,
         inputs,
         outputs,
-        test_size = None,
+        train_size = None,
         optim = 'Adam',
         init_lr = 0.01,
         crit = 'MSE',
@@ -24,12 +24,19 @@ class Trainer:
         self.batch_size = batch_size
         self.nets = nets
         self.init_lr = init_lr
-
-        self.X_train, self.X_test, self.y_train, self.y_test = sample_data (
-            inputs,
-            outputs,
-            test_size
-        )
+        if train_size is None:
+
+            self.X_train = inputs
+            self.X_test  = inputs
+            self.y_train = outputs
+            self.y_test  = outputs
+        else:
+            
+            self.X_train, self.X_test, self.y_train, self.y_test = sample_data (
+                inputs,
+                outputs,
+                min(1,np.abs(1 - train_size))
+            )
 
         # --- Fonction de perte ---
         self.crit = crit_list.get(crit)

NeuralNetworks/Trainer/train.py

@@ -27,7 +27,8 @@ def train_f (Trainer, num_epochs = 1, activate_tqdm = True):
             pbar = tqdm (
                 range (num_epochs),
                 desc = f"train epoch",
-                disable = not (activate_tqdm)
+                disable = not (activate_tqdm),
+                mininterval=0.5
             )
 
             for epoch in pbar:
@@ -43,14 +44,8 @@ def train_f (Trainer, num_epochs = 1, activate_tqdm = True):
                     def closure ():
                         Trainer.optims [k].zero_grad (set_to_none = True)
                         with autocast (dev):
-                            loss = Trainer.crit (
-                                net.f (
-                                    torch.cat (
-            [net.model (encoding (Trainer.X_train [idx]))for encoding in net.encodings],
-            dim = 1
-                                    )
-                                ),
-                                Trainer.y_train[idx]
+                            loss = Trainer.crit (net.train_forward (Trainer.X_train [idx]),
+                                Trainer.y_train [idx]
                             )
                             scaler.scale (loss).backward ()
                             return loss
@@ -66,7 +61,7 @@ def train_f (Trainer, num_epochs = 1, activate_tqdm = True):
                 for param_group in Trainer.optims [k].param_groups:
                     param_group ['lr'] = net.learnings[-1]
                 
-                pbar.set_postfix(loss=f"{epoch_loss:.5f}",lr=f"{net.learnings[-1]:.5f}")
+                #pbar.set_postfix(loss=f"{epoch_loss:.5f}",lr=f"{net.learnings[-1]:.5f}")
                 
             net = net.to ('cpu')
             net.learnings.pop(-1)

NeuralNetworks/__init__.py

@@ -15,4 +15,4 @@ from .Trainer import Trainer
 
 from .UI import *
 
-__version__ = "0.2.5"
+__version__ = "0.2.6"

NeuralNetworks/shared/__init__.py

@@ -0,0 +1,8 @@
+# NeuralNetworks - Multi-Layer Perceptrons avec encodage Fourier
+# Copyright (C) 2026 Alexandre Brun
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+from .module import Module

NeuralNetworks/shared/module.py

@@ -0,0 +1,41 @@
+# NeuralNetworks - Multi-Layer Perceptrons avec encodage Fourier
+# Copyright (C) 2026 Alexandre Brun
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+from ..Dependances import norm_list, nn, torch, device
+
+class Module (nn.Module):
+
+    def __init__ (self, name = "Net"):
+        super ().__init__ ()
+
+        # --- Attributs ---
+        self.losses = []
+        self.learnings = []
+        self.name = name
+
+    def _forward (self,x):
+        raise Exception ("_forward n'est pas défini dans la classe")
+
+    def train_forward (self,x):
+        raise Exception ("train_forward n'est pas défini dans la classe")
+            
+    def forward (self, x):
+
+        with torch.no_grad ():
+            x = x.unsqueeze (0) if x.dim () == 1 else x
+            self = self.to (device)
+            x = x.to (device)
+
+            output = self._forward(x).cpu ().numpy ().flatten ()
+            
+            x = x.to ('cpu')
+            self = self.to ('cpu')
+
+            return output
+
+            
+    

{neuralnetworks-0.2.5.dist-info → neuralnetworks-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NeuralNetworks
-Version: 0.2.5
+Version: 0.2.6
 Summary: Multi-Layer Perceptrons with multi-Fourier encoding, variable learning rate, visualization and PyTorch compilation
 Author-email: Alexandre Brun <alexandre51160@gmail.com>
 License: GPL-3.0-or-later

neuralnetworks-0.2.6.dist-info/RECORD

@@ -0,0 +1,22 @@
+NeuralNetworks/__init__.py,sha256=DFqmUzb9LQDti9PGdZgvR-1irNeurxsVhvwPEuBj-ro,668
+NeuralNetworks/Dependances/__init__.py,sha256=qEpDbSD8cCq-E5XVisNUVf3kZOYopDnQWToyRefPgKE,1227
+NeuralNetworks/Dependances/matplot.py,sha256=elS8u6DZHYP-8mHEpYNOw3jDzhCAWTld9tm3OAD46zw,957
+NeuralNetworks/Dependances/pytorch.py,sha256=r07s7EYeSQJh0PqODaMby4RttZf8m1mUyHrM3uXm370,3287
+NeuralNetworks/MLP/FourierFeatures.py,sha256=klgRM1HK09oA2NRMDxQMjJJ-WoUd5hV1ip5hHe9rHjI,3250
+NeuralNetworks/MLP/Layers.py,sha256=WAksXsiMxaClyYTxPhlyQbwwj9qTtXs3EWCO1RqjUHY,945
+NeuralNetworks/MLP/__init__.py,sha256=zQUczdC-5xkMBZ-cOFpHJrbgJz0grmVRABEdqk7080c,2739
+NeuralNetworks/MLP/inference.py,sha256=pZja1yhkT3BUh9E2X__2LO5QuAamPHITI32rT5Q_cQo,543
+NeuralNetworks/Trainer/__init__.py,sha256=XYfwidMxpI30y8j3FUgvhtRMrrWLvszZE2EwM72tFq4,1952
+NeuralNetworks/Trainer/dynamic_learning_rate.py,sha256=1JAD-k0cjdL_71zGeeCUFOa61H4PzFITDjZ2nK0TzXU,2340
+NeuralNetworks/Trainer/sample_data.py,sha256=7waC9colb7DXU4yKMcgcCnPG3Guv-isipcgVHJPPCNE,673
+NeuralNetworks/Trainer/train.py,sha256=dpo1VDu0VNlzXPqJWGkmcY_y18Y05DTjA5wDg7AbLco,2989
+NeuralNetworks/UI/Learnings.py,sha256=4TBR5pcjyoBeL7eikNKM6xn25jnqL-mWT7hbrt9q-Gw,1418
+NeuralNetworks/UI/Losses.py,sha256=Tu5xuDiutR9a4xcZKpyWN_tzSDu3_fImEf8FbAEehio,1378
+NeuralNetworks/UI/__init__.py,sha256=L96xwQZJ-HoqqOGxaheosiDKHR3mRopuXkif--rO1J4,409
+NeuralNetworks/shared/__init__.py,sha256=LKnIP46kbzOy7O3bSALwUg7iu8_wpis0LzUVVaW20Ws,376
+NeuralNetworks/shared/module.py,sha256=Y6A1LJRFPWOhwNOjPTTyzxPVafEZMbko_fLXKtXoDYI,1196
+neuralnetworks-0.2.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+neuralnetworks-0.2.6.dist-info/METADATA,sha256=1p55SI9TCwSSkj6N1bR0AgXOWA5fFKUYs1ZPlvSUx40,18349
+neuralnetworks-0.2.6.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+neuralnetworks-0.2.6.dist-info/top_level.txt,sha256=h18nmC1BX7avyAAwKh0OQWezxgXmOpmVtbFq-8Mcbms,15
+neuralnetworks-0.2.6.dist-info/RECORD,,

{neuralnetworks-0.2.5.dist-info → neuralnetworks-0.2.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

neuralnetworks-0.2.5.dist-info/RECORD

@@ -1,20 +0,0 @@
-NeuralNetworks/__init__.py,sha256=DSCQD_dezpOEjqAGUSQsocNYmNq8STPHZaLjIXXLCF8,668
-NeuralNetworks/Dependances/__init__.py,sha256=qEpDbSD8cCq-E5XVisNUVf3kZOYopDnQWToyRefPgKE,1227
-NeuralNetworks/Dependances/matplot.py,sha256=elS8u6DZHYP-8mHEpYNOw3jDzhCAWTld9tm3OAD46zw,957
-NeuralNetworks/Dependances/pytorch.py,sha256=RQlSV3-8uHAoEgK0FBae7O4Mdug7h_MY--sN1fK59qw,3329
-NeuralNetworks/MLP/FourierFeatures.py,sha256=klgRM1HK09oA2NRMDxQMjJJ-WoUd5hV1ip5hHe9rHjI,3250
-NeuralNetworks/MLP/Layers.py,sha256=WAksXsiMxaClyYTxPhlyQbwwj9qTtXs3EWCO1RqjUHY,945
-NeuralNetworks/MLP/__init__.py,sha256=v7h4Vcmay0_A83spe67HhT7TQ7cSKQX4eVVBJhzEWfk,3052
-NeuralNetworks/MLP/inference.py,sha256=9aL7pUx1LTVvrc6UYHX049UjODTgHY6cweFcp2gequQ,853
-NeuralNetworks/Trainer/__init__.py,sha256=v0qKqx9XkYWkuouNNy0jTHQ_cZqYhFj98qrwSXlDXy0,1711
-NeuralNetworks/Trainer/dynamic_learning_rate.py,sha256=1JAD-k0cjdL_71zGeeCUFOa61H4PzFITDjZ2nK0TzXU,2340
-NeuralNetworks/Trainer/sample_data.py,sha256=7waC9colb7DXU4yKMcgcCnPG3Guv-isipcgVHJPPCNE,673
-NeuralNetworks/Trainer/train.py,sha256=NAbHFKg4hl96OXq_i63lcRYwrPHiuKu7ihexakhpgDY,3182
-NeuralNetworks/UI/Learnings.py,sha256=4TBR5pcjyoBeL7eikNKM6xn25jnqL-mWT7hbrt9q-Gw,1418
-NeuralNetworks/UI/Losses.py,sha256=Tu5xuDiutR9a4xcZKpyWN_tzSDu3_fImEf8FbAEehio,1378
-NeuralNetworks/UI/__init__.py,sha256=L96xwQZJ-HoqqOGxaheosiDKHR3mRopuXkif--rO1J4,409
-neuralnetworks-0.2.5.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-neuralnetworks-0.2.5.dist-info/METADATA,sha256=FjTIFWlGmzIjQmWhFLnTuUI7MfNl0U5jb7EjRQJ7lh8,18349
-neuralnetworks-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuralnetworks-0.2.5.dist-info/top_level.txt,sha256=h18nmC1BX7avyAAwKh0OQWezxgXmOpmVtbFq-8Mcbms,15
-neuralnetworks-0.2.5.dist-info/RECORD,,