【发布时间】:2015-11-15 20:25:21
【问题描述】:
我该如何解决这个MemoryError问题?
我在train3.csv 中有 642,709 行
.train() 调用失败。
我有 4GB 的 DDR3 内存。
有没有办法让 MemoryError 不会失败,就像其他训练方法或以某种方式增加我的虚拟内存(我在 Windows 10 上)?
代码:
train_file = 'train3.csv'
netsave_file = 'neurolab.net'
hidden_units = 440
outputs = 1
import numpy as np
import neurolab as nl
# read training data and put it into numpy array _______________________
t = []
t_file = open(train_file, 'r')
for line in t_file.readlines():
train = line.split(',')
train[1] = int(train[1])
for i in range(0,72):
train[i+2] = float(train[i+2]) # convert to floats
t.append(train)
t_file.close()
print "training samples read: " + str(len(t))
input = []
target = []
for train in t:
input.append(train[2:2+72])
target.append(train[1:2])
print "done reading input and target"
train = 0
input = np.array(input)
target = np.array(target)
print "done converting input and target to numpy array"
net = nl.net.newff([[0.0,1.0]]*72, [hidden_units,144,outputs])
# Train process _______________________________________________________
err = net.train(input, target, show=1, epochs = 2)
net.save(netsave_file)
显示此错误:
Traceback (most recent call last):
File "neurolab_train.py", line 43, in <module>
err = net.train(input, target, show=1, epochs = 2)
File "C:\Users\tintran\Anaconda\lib\site-packages\neurolab\core.py", line 165, in train
return self.trainf(self, *args, **kwargs)
File "C:\Users\tintran\Anaconda\lib\site-packages\neurolab\core.py", line 349, in __call__
train(net, *args)
File "C:\Users\tintran\Anaconda\lib\site-packages\neurolab\train\spo.py", line 79, in __call__
**self.kwargs)
File "C:\Users\tintran\Anaconda\lib\site-packages\scipy\optimize\optimize.py", line 782, in fmin_bfgs
res = _minimize_bfgs(f, x0, args, fprime, callback=callback, **opts)
File "C:\Users\tintran\Anaconda\lib\site-packages\scipy\optimize\optimize.py", line 840, in _minimize_bfgs
I = numpy.eye(N, dtype=int)
File "C:\Users\tintran\Anaconda\lib\site-packages\numpy\lib\twodim_base.py", line 231, in eye
m = zeros((N, M), dtype=dtype)
MemoryError
【问题讨论】:
标签: python neural-network