【发布时间】:2013-09-25 20:17:00
【问题描述】:
当我使用 partial_fit 方法训练 SGDClassifier 时,我正在尝试拼凑它如何选择学习率。
即,我的主要学习循环如下所示:
from sklearn.linear_model import SGDClassifier
m = SGDClassifier(n_iter=1, alpha=0.01)
n_iter = 40
t0 = time.time()
for i in range(n_iter):
for fname in files:
X, y = load_next_batch(fname)
m.partial_fit(X, y, classes = [0, 1])
print "%d: valid-error: %f (time: %fs)" % (i, 1.0-m.score(Xvalid, yvalid), time.time() - t0)
现在,由于我对整个训练集进行了 40 次传递,因此我想随着时间的推移调整我的学习率。如果我使用fit 而不是部分拟合,我的理解是这会自动发生(除非我修改learning_rate 参数)。
但是,我不清楚在使用部分拟合时会发生这种情况。略读代码也无济于事。谁能澄清我如何才能达到退火学习率 在我的设置中?
【问题讨论】:
标签: scikit-learn gradient-descent