神经网络python库_Python神经网络库PyBrain简单使用

关键字：python、pybrain、神经网络

时间：2016年12月

前言

pybrain，一个基于python的神经网络库。

代码

# -*- coding: utf-8 -*-

from pybrain.datasets import SupervisedDataSet

from pybrain.supervised.trainers import BackpropTrainer

from pybrain.tools.shortcuts import buildNetwork

#创建一个有3个输入、4个隐层和1个输出的神经网络

net = buildNetwork(3, 4, 1)

#生成样本

ds = SupervisedDataSet(3, 1)

ds.addSample((0, 0, 1), (0))

ds.addSample((1, 1, 1), (1))

ds.addSample((1, 0, 1), (1))

ds.addSample((0, 1, 1), (0))

#生成训练器

trainer = BackpropTrainer(net, ds)

#指定训练次数方式训练

#for i in range(0,10000):

# trainer.train()

#自动控制训练次数方式训练

trainer.trainUntilConvergence()

#测试

print '0,0,1 0', net.activate([0,0,1])

print '1,1,1 1', net.activate([1,1,1])

print '1,0,1 1', net.activate([1,0,1])

print '0,1,1 0', net.activate([0,1,1])