def fit(self,X,y):
self.coef_ = None
self.intercept_ = None
self.stages_ = []
for i in range(0,self.max_iter,self.n_iter):
if self.coef_ is not None:
assert(self.intercept_ is not None)
linear_model.SGDRegressor.fit(self,X,y,coef_init=self.coef_,intercept_init=self.intercept_)
else:
linear_model.SGDRegressor.fit(self,X,y)
# record coefs and intercept for later
self.stages_.append((i+self.n_iter,self.coef_.copy(),self.intercept_.copy()))
logging.info('done %d/%d steps' % (i+self.n_iter,self.max_iter))
logging.info('training set auc %f' % self.auc(X,y))
评论列表
文章目录