Как передать параметры этой модели Sklearn Cox в конвейере? Я получаю сообщение об ошибке ⇐ Python

[Anonymous]

Если я запускаю следующий код Python, он работает хорошо:

Код: Выделить всё

tranOH = ColumnTransformer([ ('one', OneHotEncoder(drop='first', dtype='int'),
make_column_selector(dtype_include='category', pattern=f"^(?!{target}).*")
) ], remainder='passthrough')

dftrain2 = tranOH.fit_transform(dftrain)
cph = CoxPHFitter(penalizer=0.1)
cph.fit(dftrain2, 'months', 'churn')

Но если я попытаюсь сделать это с помощью конвейера, я получу ошибку:

Код: Выделить всё

mcox = Pipeline(steps=[
("onehot", tranOH),
('modelo', CoxPHFitter(penalizer=0.1))
])

mcox.fit(dftrain, modelo__duration_col="months", modelo__event_col='churn')

В нем говорится:

CoxPHitter.fit() получил несколько значений для аргумента 'duration_col'
tranOH — это преобразователь столбцов, который onehot кодирует все категориальные столбцы, кроме «оттока».
Я также пробовал использовать col="months" и event_col=target непосредственно внутри CoxPHitter(), но я получаю ту же ошибку.
Позже я хочу использовать его для выполнения GridSearchCV для точно настройте параметр пенализатора, оптимизируя показатель точности для прогнозирования оттока через несколько месяцев.

Подробнее здесь: https://stackoverflow.com/questions/790 ... t-an-error