2012-04-25 5 views
5

есть функция: y = sin (x) Я хочу использовать сети PyBrain для соответствия функциям, вот что я сделал: , когда вы запустите его, вы получите то, что я получаю, полученные данные далеки от чего это должно быть.Как установить функцию с помощью сетей PyBrain?

from pybrain.datasets import SupervisedDataSet 
from pybrain.tools.shortcuts import buildNetwork 
from pybrain.supervised.trainers import BackpropTrainer 
import pickle 
import scipy as sp 
import numpy as np 
import pylab as pl 

x = np.linspace(0, 4*np.pi, 100) 
ds = SupervisedDataSet(1,1) 

for i in x: 
    ds.addSample(i,sin(i)) 
print ds 

n = buildNetwork(ds.indim,3,3,3,ds.outdim,recurrent=True) 
t = BackpropTrainer(n,learningrate=0.01,momentum=0.5,verbose=True) 
t.trainOnDataset(ds,1000) 
t.testOnData(verbose=True) 

fileObject = open('trained_net', 'w') 
pickle.dump(n, fileObject) 
fileObject.close() 

fileObject = open('trained_net','r') 
net = pickle.load(fileObject) 

y = [] 
for i in x: 
    y.append(net.activate(i)) 

pl.plot(x,y) 
pl.plot(x,np.sin(x)) 
pl.show() 
+0

Итак, в чем ваш вопрос? Пробовали ли вы другие сетевые архитектуры, чем этот 5-слойный пример? – schaul

ответ

8

Я полагаю, ваша проблема в том, что эта сеть не подходит для этой функции. Общее количество сетевых узлов слишком низкое для правильной установки этой функции sin (x): функция слишком сложна. Кроме того, для установки любой функции в принципе требуется не более одного скрытого слоя.

Например, попробуйте удалить два скрытых слоя и увеличить количество скрытых узлов (например, 20). Ваш код отлично подходит для функции

Смежные вопросы