Уменьшить размер модели авто-склеарн

Question

Я использую auto-sklearn для генерации регрессионной модели на основе некоторых данных. После нескольких часов работы я сохраняю сгенерированную модель на диск для последующего использования с joblib, и размер сгенерированного файла составляет 2,5 ГиБ.

Как я могу уменьшить размер файла сохраненной модели? Мне нужно только иметь возможность делать прогнозы с моделью в будущем.

Answer 1

В зависимости от того, какую модель вы используете, существует большая вероятность, что вы абсолютно не сможете. Если у вас такая большая модель, я думаю, это модель нейронной сети или случайного леса. И, к сожалению, не существует простого способа сделать это, и если вы сделаете это, вы, скорее всего, снизите свою точность.

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

Если ваш вопрос был: есть ли в модели что-то полезное только для обучения, которое я могу удалить => может быть, несколько переменных, ничего достаточно большого, чтобы стоить вашего времени (максимум несколько КБ).