Определение стоимости автомобилей
Сервис по продаже автомобилей с пробегом «Не бит, не крашен» разрабатывает приложение для привлечения новых клиентов. В нём можно быстро узнать рыночную стоимость своего автомобиля. В вашем распоряжении исторические данные: технические характеристики, комплектации и цены автомобилей. Вам нужно построить модель для определения стоимости.
Заказчику важны:
- качество предсказания;
- скорость предсказания;
- время обучения.
Инструкция по выполнению проекта
Чтобы усилить исследование, не ограничивайтесь градиентным бустингом. Попробуйте более простые модели — иногда они работают лучше. Эти редкие случаи легко пропустить, если всегда применять только бустинг.
Поэкспериментируйте и сравните характеристики моделей: время обучения, время предсказания, точность результата.
Основные шаги:
- Загрузите данные, путь к файлу: /datasets/autos.csv.
- Изучите данные. Заполните пропущенные значения и обработайте аномалии в столбцах. Если среди признаков имеются неинформативные, удалите их.
- Подготовьте выборки для обучения моделей.
- Обучите разные модели, одна из которых — LightGBM, как минимум одна — не бустинг. Для каждой модели попробуйте разные гиперпараметры.
- Проанализируйте время обучения, время предсказания и качество моделей.
- Опираясь на критерии заказчика, выберете лучшую модель, проверьте её качество на тестовой выборке.
Примечания:
- Для оценки качества моделей применяйте метрику RMSE.
- Значение метрики RMSE должно быть меньше 2500.
- Самостоятельно освойте библиотеку LightGBM и её средствами постройте модели градиентного бустинга.
- Время выполнения ячейки кода Jupyter Notebook можно получить специальной командой. Найдите её.
- Модель градиентного бустинга может долго обучаться, поэтому измените у неё только два-три параметра.
- Если перестанет работать Jupyter Notebook, удалите лишние переменные оператором
del.
Описание данных
Данные находятся в файле /datasets/autos.csv.
Признаки
-
DateCrawled— дата скачивания анкеты из базы -
VehicleType— тип автомобильного кузова -
RegistrationYear— год регистрации автомобиля -
Gearbox— тип коробки передач -
Power— мощность (л. с.) -
Model— модель автомобиля -
Kilometer— пробег (км) -
RegistrationMonth— месяц регистрации автомобиля -
FuelType— тип топлива -
Brand— марка автомобиля -
Repaired— была машина в ремонте или нет -
DateCreated— дата создания анкеты -
NumberOfPictures— количество фотографий автомобиля -
PostalCode— почтовый индекс владельца анкеты (пользователя) -
LastSeen— дата последней активности пользователя
Целевой признак
-
Price— цена (евро)