关键字: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])