Pipeline

Конвейер из последовательных шагов обработки и модели.

СигнатураPipeline(steps=[('name', estimator), ...])

Объединяет предобработку и модель в единый объект: при вызове fit каждый шаг последовательно обучается. Предотвращает утечку данных при кросс-валидации и упрощает деплой.

from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LogisticRegression

pipe = Pipeline([
    ('scaler', StandardScaler()),
    ('clf', LogisticRegression()),
])
pipe.fit(X_train, y_train)
print(pipe.score(X_test, y_test))
← Все записи: scikit-learn
Поддержать проект