#python
#python
Вопрос:
Я пытаюсь сохранить лучшую модель после использования KFold. Есть ли какой-либо способ сохранить лучшую модель в зависимости от f1_score каждого сгиба?
Комментарии:
1. О какой модели вы говорите.?
2. Это модель LSTM @SreeramTP
3. ваш вопрос такой расплывчатый. Какую библиотеку вы используете.? Keras.? TF.? Pytorch.?
4. Я все равно использую Keras, я нашел способ решить эту проблему. Спасибо, что помогли мне. Я оставлю ответ на случай, если кто-нибудь захочет знать.
Ответ №1:
Используйте библиотеку PICKLE в python, чтобы сохранить вашу модель в виде файла .pkl. Вы можете повторно использовать этот файл модели, когда захотите. Это постоянная модель.
Пожалуйста, дайте мне знать, если у вас возникнут какие-либо вопросы. Я был бы очень рад вам помочь.
Комментарии:
1. Да, я знаю, что Pickle может помочь мне сохранить модель в виде файла, но я хочу сохранить только лучшую модель с самым высоким f1_score в каждом сгибе, пока вы запускаете model
2. Это зависит от используемой вами библиотеки. Если вы используете библиотеку Sklearn и для решения задачи логистической регрессии, используйте метод GridSearchCV, где нам нужно передать CV (перекрестные проверочные сгибы), параметры сетки (C, штраф — L1 (Лассо), L2 (гребень)). Чтобы вы могли подогнать модель под эти гиперпараметрические оптимизации и сохранить с помощью pickle. Пожалуйста, дайте мне знать, если вам нужен какой-либо пример кода, я был бы очень рад предоставить.