diff --git a/README.md b/README.md index 27a0880..1ed4c56 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,9 @@ # PT21-22-Reseau-Neurones +Cette librairie python permet de créer simplement un modèle de perceptron multicouche en choisissant les hyperparamètres suivants: +Le nombre de couches +Le nombre de neurones pour chaque couche +Le learning rate +Le batch size +Le nombre d’époques +Il est également possible de donner un jeu de données supplémentaire afin de mesurer la précision du modèle à la fin de l'entraînement. Lorsqu’il n'y a que deux neurones d'entrées, il y a une option pour visualiser l’entraînement. \ No newline at end of file diff --git a/tests/MNISTDrawingPrediction.py b/tests/MNISTDrawingPrediction.py index 2a65a7c..70c4981 100644 --- a/tests/MNISTDrawingPrediction.py +++ b/tests/MNISTDrawingPrediction.py @@ -14,7 +14,7 @@ class Sketchpad(tkinter.Canvas): self.bind("", self.add_line) self.PILImage = Image.new("F", (560, 560), 100) self.draw = ImageDraw.Draw(self.PILImage) - self.MNISTNN = network.networkFromFile("MNISTtest2") + self.MNISTNN = network.networkFromFile("MNIST30epoch") self.predictionLabel = predictionLabel def add_line(self, event):