Neural Network training with PyBrain won't converge

前端 未结 4 1672
迷失自我
迷失自我 2020-12-08 04:36

I have the following code, from the PyBrain tutorial:

from pybrain.datasets import SupervisedDataSet
from pybrain.supervised.trainers import BackpropTrainer
         


        
4条回答
  •  情话喂你
    2020-12-08 05:05

    trainer = BackpropTrainer(net, ds, learningrate = 0.9, momentum=0.0, weightdecay=0.0, verbose=True) 
    trainer.trainEpochs(epochs=1000)
    

    This way can converge. if learningrate is too small(e.g. 0.01), it lost in local minimum. As I have tested, learningrate in 0.3-30, it can converge.

提交回复
热议问题