[PYTHON] Exporter et importer des modèles formés avec PyBrain

Récemment, j'ai utilisé PyBrain, une bibliothèque d'apprentissage automatique pour Python, pour créer et entraîner des réseaux de neurones. Avec PyBrain, il est assez facile de créer un réseau neuronal et vous pouvez exporter le modèle entraîné sous forme de fichier xml. Bien sûr, si vous lisez ceci, vous pouvez réutiliser le modèle.

Donc, cette fois je vais résumer la méthode d'installation et l'écriture et le chargement du modèle.

1. Installez PyBrain

Utilisez pip pour l'installation.

pip install pybrain

2. Exporter le modèle

Utilisez from pybrain.tools.xml import NetworkWriter.

from pybrain.tools.shortcuts import buildNetwork

# ~Omission(Lecture des données d'entraînement, etc.) ~

#Définition du modèle de réseau neuronal
network = buildNetwork(64, 19, 2)

# ~Omission (effectuer un entraînement à l'aide de données)~

#Exporter le modèle
NetworkWriter.writeToFile(network, 'model.xml')

Cela produira un fichier appelé model.xml avec les paramètres du modèle entraîné.

3. Chargez le modèle

Utilisez from pybrain.tools.xml import Network Reader.

from pybrain.tools.xml import NetworkReader

network = NetworkReader.readFrom('model.xml')

En lisant la sortie model.xml plus tôt, le modèle entraîné peut être reproduit.

Après cela, veuillez le reconnaître, en savoir plus ou l'aimer.