Как преобразовать datapoint для предсказания в sklearn

130
16

Я построил модель в sklearn, которую я сохранил в форме рассола. Однако, чтобы получить наилучшие результаты, которые я смог получить, мне пришлось удалить функции, создать другие, использовать горячую кодировку и т.д.

Теперь, я получил новый datapoint в первоначальной форме, и мне нужно сделать прогноз. Есть ли простой способ преобразовать datapoint, так что моя построенная модель может сделать прогноз?

спросил(а) 2021-01-19T15:15:22+03:00 9 месяцев назад
1
Решение
65

Вам необходимо применить те же преобразования к тестовому набору, что и для тренировки набора. Один из способов сделать это - написать функцию для обработки данных и использовать ее. Итак, у вас будет что-то вроде этого:

def process_data(x):
# process data, remove features and so on

return processed_data

X_train_processed = process(X_train)
X_test_processed = process(X_test)

ответил(а) 2021-01-19T15:15:22+03:00 9 месяцев назад
Ваш ответ
Введите минимум 50 символов
Чтобы , пожалуйста,
Выберите тему жалобы:

Другая проблема