[PYTHON] Exportieren und importieren Sie geschulte Modelle mit PyBrain

Kürzlich habe ich PyBrain, eine Bibliothek für maschinelles Lernen für Python, verwendet, um neuronale Netze aufzubauen und zu trainieren. Mit PyBrain ist es ziemlich einfach, ein neuronales Netz aufzubauen, und Sie können das trainierte Modell als XML-Datei exportieren. Wenn Sie dies lesen, können Sie das Modell natürlich wiederverwenden.

Dieses Mal werde ich die Installationsmethode zusammenfassen und das Modell schreiben / laden.

1. Installieren Sie PyBrain

Verwenden Sie pip für die Installation.

pip install pybrain

2. Modell exportieren

Verwenden Sie from pybrain.tools.xml import NetworkWriter.


from pybrain.tools.shortcuts import buildNetwork

# ~Unterlassung(Trainingsdaten lesen usw.) ~

#Definition des neuronalen Netzwerkmodells
network = buildNetwork(64, 19, 2)

# ~Auslassung (Training mit Daten durchführen)~

#Exportieren Sie das Modell
NetworkWriter.writeToFile(network, 'model.xml')

Dadurch wird eine Datei mit dem Namen model.xml mit den Einstellungen des trainierten Modells ausgegeben.

3. Laden Sie das Modell

Verwenden Sie from pybrain.tools.xml import Network Reader.

from pybrain.tools.xml import NetworkReader

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

Durch früheres Lesen der Ausgabe von model.xml kann das trainierte Modell reproduziert werden.

Danach erkennen Sie es bitte, erfahren Sie mehr oder mögen Sie es.

Recommended Posts

Exportieren und importieren Sie geschulte Modelle mit PyBrain
Excel, CSV Import / Export mit Django
So exportieren / importieren Sie ein Zope-Objekt von Zope, das mit Plone-4.1.5-UnifiedInstaller-20120604.tgz installiert wurde
Importieren Sie tsv mit Python
Importieren Sie vtk mit Brew Python
PyTorch mit AWS Lambda [Lambda-Import]
Serververwaltung mit Jupyter (1) -Import