Поддержка Xgboost и LSTM в Apache Ignite ML и DL - PullRequest
0 голосов
/ 14 мая 2018

В нашей системе машинного обучения мы использовали алгоритмы Xgboost и LSTM.Я хочу использовать библиотеки машинного обучения на основе Grid Apache Ignite In-Memory ( Apache Ignite Machine Learning Grid ) для оптимизации производительности этой системы.Поддерживают ли библиотеки Apache Ignite ML и DL алгоритмы Xgboost и LSTM?

1 Ответ

0 голосов
/ 14 мая 2018

Зависит от вашего местоположения данных и ваших целей.Ваши данные уже хранятся в Apache Ignite?Используете ли вы Apache Ignite в своей экосистеме?

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

Что касается LSTM , Apache Ignite ML в настоящее время поддерживает только нейронные сети MLP, поэтому вы не можете обучить свою модель в Apache Ignite.Тем не менее, вы все равно можете обучить модель за пределами Apache Ignite и затем импортировать подготовленную модель.

...