Профилирование нестандартного склеарн конвейера - PullRequest
Новая
       40

Профилирование нестандартного склеарн конвейера

1 голос
/ 22 апреля 2020

Я создал конвейер sklearn для моего проекта с двумя компонентами, а именно data_transform и model, как показано ниже. 

model = Pipeline([
    ('data_transform', XYZ(p1=arg1, p2=arg2)),
    ('model', LogisticRegressionCV(solver='sag', multi_class='multinomial', class_weight='balanced', max_iter=5000))])

Я называю метод fit как model.fit(X_train, y_train). Поскольку мой код занимает много времени, я хотел проверить, сколько времени занимает каждый компонент, то есть data_transform и model. Есть какой-нибудь метод, из которого я мог бы найти время, взятое?

...