我正在尝试使用一个partial_fit
在小批量上的SGDClassifier来训练逻辑回归模型:
sgdLogReg = SGDClassifier(loss='log')
for i in range(math.ceil(len(X_train)/mini_batch_size)):
sgdLogReg.partial_fit(X_train[i*mini_batch_size:(i+1)*mini_batch_size],
Y_train[i*mini_batch_size:(i+1)*mini_batch_size],
classes=[0,1])
但是,由于我没有使用fit
函数,因此未设置sgdLogReg.coef_
属性:
AttributeError: 'SGDClassifier' object has no attribute 'coef_'
有没有办法用partial_fit
得到系数向量
我想,您是在拟合数据之前调用
coef_
。下面的示例显示,您可以在.partial_fit
之后获取coef_
相关问题 更多 >
编程相关推荐