Есть ли функция ввода-вывода для хранения обученных моделей в кедро?
В разделе IO документации kedro API я не смог найти функциональность по хранению обученных моделей (например,.pkl, .joblib, ONNX, PMML)? Я что-то упустил?
2 ответа
Здесь pickle набор данных в kedro.io, который можно использовать для сохранения обученных моделей и / или всего, что вы хотите обработать, и который можно сериализовать (модели являются общим объектом). Он принимаетbackend это по умолчанию pickle но может быть установлен на joblib если вы хотите использовать joblib вместо.
Замечу лишь, что Кедро переезжает в kedro.extras.datasets для своих наборов данных и отказ от дополнительных наборов данных в kedro.io. Вы можете посмотреть наkedro.extras.datasets и в Kedro 0.16 и далее pickle.PickleDataSet с joblib поддержка.
Кедро spaceflights учебник в документации фактически сохраняет обученную модель линейной регрессии с помощью pickleнабор данных, если вы хотите увидеть его пример. Соответствующий раздел здесь.
PickleDataSet есть в https://kedro.readthedocs.io/en/latest/kedro.extras.datasets.pickle.PickleDataSet.html, а поддержка joblib в PickleDataSet будет в следующем выпуске (см. https://github.com/quantumblacklabs/kedro/blob/develop/RELEASE.md)