Гайд · TNWS AI

Как обучить MLPClassifier со StandardScaler без утечки данных

4 мин—

Проверяемый Pipeline из StandardScaler и MLPClassifier: раздельная выборка, обучение scaler только на train, accuracy и защита от утечки.

Задача и применимость

Актуальный запрос: «как обучить нейросеть MLPClassifier StandardScaler Pipeline». Собрать минимальную полносвязную нейросеть для табличной бинарной классификации и не подмешать статистики тестовой выборки в масштабирование. MLP чувствителен к масштабу признаков: большие по модулю столбцы способны доминировать в оптимизации. Pipeline обучает StandardScaler на train, преобразует данные и передаёт их MLPClassifier. При predict тот же scaler применяется автоматически.

Материал относится к scikit-learn 1.8.0 и табличным числовым данным. MLPClassifier — полносвязный многослойный перцептрон для обучения с учителем, а не готовая языковая или vision-модель. Все вычисления ниже выполняются локально на синтетическом наборе make_classification: интернет, ключи API и внешние персональные данные не нужны. Значения метрик в рабочем проекте будут другими, поэтому фиксированные пороги примера являются проверкой кода, а не обещанием качества.

Пошаговое решение

  1. Создайте воспроизводимый синтетический набор и стратифицированный train/test split.
  2. Объедините StandardScaler и MLPClassifier через make_pipeline.
  3. Обучите только на X_train и y_train.
  4. Посчитайте accuracy на отложенной выборке и проверьте среднее scaler по train.

Готовый проверяемый шаблон

from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split

X, y = make_classification(
    n_samples=500, n_features=8, n_informative=6, n_redundant=0,
    class_sep=1.4, random_state=42
)
X_train, X_test, y_train, y_test = train_test_split(
    X, y, test_size=0.25, stratify=y, random_state=42
)
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.neural_network import MLPClassifier

model = make_pipeline(
    StandardScaler(),
    MLPClassifier(hidden_layer_sizes=(16,), solver="lbfgs", max_iter=500, random_state=42)
)
model.fit(X_train, y_train)
accuracy = model.score(X_test, y_test)
scaler = model.named_steps["standardscaler"]
assert scaler.mean_.shape == (8,)
assert accuracy > 0.80
print(round(accuracy, 3))

Вход и ожидаемый результат

Вход — 500 строк с восемью числовыми признаками и бинарной меткой. В тестовой установке фиксируются random_state и стратификация. Ожидается accuracy выше 0,80, массив mean_ длины восемь и возможность вызвать model.predict на сырых, ещё не масштабированных строках.

Как проверить результат

Сравните scaler.mean_ со средним только X_train: они должны совпадать. Среднее всего X при честном split обычно отличается. Это прямой тест отсутствия leakage. Затем сохраните индексы разбиения и повторите прогон.

Перенос на рабочую нейросеть

Заменяя синтетические данные рабочими, сначала составьте контракт: одна строка соответствует одному объекту, столбцы имеют стабильный порядок и единицы измерения, target формируется без информации из будущего. Категориальные поля кодируются отдельным преобразованием, пропуски обрабатываются до StandardScaler совместимым способом. Объединяйте preprocessing и estimator в Pipeline или ColumnTransformer, чтобы одинаковая логика применялась при fit, validation и predict. Версионируйте список признаков, код подготовки, разбиение и сам estimator как единый выпуск.

Критерии приёмки

  1. Train, validation и test не пересекаются, а random_state и алгоритм разбиения записаны.
  2. Preprocessing обучается только на train и входит в сохраняемый Pipeline.
  3. Форма, dtype и порядок признаков проверяются до predict.
  4. Метрика считается на независимой выборке и сопровождается support.
  5. Повторный запуск в закреплённом окружении воспроизводит контрольные predictions.
  6. Негативный тест с лишним или отсутствующим признаком завершается явной ошибкой, а не молчаливой перестановкой.

Регрессионный протокол

Сохраните маленький обезличенный regression-набор и ожидаемые классы либо вероятности. После изменения параметров сравнивайте не только одну итоговую метрику, но и confusion matrix, время обучения, число итераций и распределение вероятностей. Меняйте один фактор за эксперимент. Гиперпараметры выбирайте на validation или внутри кросс-валидации; test открывайте только для финальной оценки выбранной процедуры. Если обучающих данных мало, публикуйте разброс по нескольким разбиениям, а не лучший результат.

Диагностика расхождений

Если качество резко изменилось, проверяйте по порядку: набор и target, индексы split, порядок столбцов, статистики preprocessing, версии зависимостей, random_state и только затем внутренние параметры нейросети. Убедитесь, что признаки не содержат идентификатор ответа или данные из будущего. При расхождении после загрузки сравните predictions на одном сохранённом массиве, classes_, число входных признаков и параметры scaler. Не лечите неизвестную причину увеличением сети: это усложняет диагностику и часто усиливает переобучение.

Контроль после внедрения

Для эксплуатации добавьте журнал версии модели и схемы входа, но не записывайте в него исходные персональные признаки. До выкладки задайте допустимые границы latency, доли ошибок и целевой метрики. После выкладки сравнивайте распределения признаков и прогнозов с контрольным периодом. Если входная схема изменилась или качество вышло за порог, переключайтесь на предыдущий проверенный артефакт; автоматическое дообучение без независимой оценки не является безопасным исправлением.

Типичные ошибки и чего не делать

  • Не вызывайте fit_transform на полном X до train_test_split.
  • Не масштабируйте train и test разными экземплярами scaler.
  • Не оценивайте качество на данных, использованных в fit.

Не подбирайте архитектуру и пороги по тестовой выборке. Не передавайте в модель персональные или чувствительные поля без правового основания и необходимости. Не считайте высокую метрику на синтетическом примере доказательством готовности к эксплуатации. Для важных решений добавьте человеческую проверку, мониторинг drift и безопасный сценарий отказа.

FAQ

Зачем Pipeline?

Он связывает preprocessing и модель в один проверяемый estimator.

Почему StandardScaler?

Он центрирует признаки и приводит их к сопоставимому масштабу.

Можно подавать сырые данные в predict?

Да, если вызывается predict у всего обученного Pipeline.

Официальные источники

Частые вопросы

Зачем Pipeline?

Он связывает preprocessing и модель в один проверяемый estimator.

Почему StandardScaler?

Он центрирует признаки и приводит их к сопоставимому масштабу.

Можно подавать сырые данные в predict?

Да, если вызывается predict у всего обученного Pipeline.

Читайте также

Комментарии

Пока тихо. Скажите первое слово