Получение коэффициентов частичного подбора склеарна SGDClassifier - PullRequest
1 голос
/ 15 января 2020

Я пытаюсь обучить регрессионную модель логистики c, используя SGDClassifier , который я partial_fit на мини-партиях:

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' нет атрибута 'coef _'

Есть ли способ получить вектор коэффициентов, используя partial_fit?

1 Ответ

0 голосов
/ 16 января 2020

Я думаю, вы звоните coef_, прежде чем подгонять данные. Пример ниже показывает, что вы можете получить coef_ после .partial_fit.

from sklearn.datasets import make_classification

X, y = make_classification()

sgd = SGDClassifier()

sgd.partial_fit(X,y, classes=[0,1])

print(sgd.coef_)

# [[-42.53636525  -2.52884548 -39.11102068 -14.67123492  16.20897739
#   -35.94926059  12.62473266  18.07587488  49.73232778 -16.74330257
#    78.61713294 -22.21668162  14.89044933 -16.34974837   6.38661036
#    18.58266319  18.84123791  28.24578991  37.94146501  21.3143532 ]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...