2020-10-26 10:29:30 +00:00
---
2022-04-09 13:29:05 +00:00
sidebar_position: 41
sidebar_label: "Применение модели CatBoost в ClickHouse"
2020-10-26 10:29:30 +00:00
---
2019-08-16 06:54:18 +00:00
# Применение модели CatBoost в ClickHouse {#applying-catboost-model-in-clickhouse}
2019-10-14 12:39:42 +00:00
[CatBoost ](https://catboost.ai ) — открытая программная библиотека разработанная компанией [Яндекс ](https://yandex.ru/company/ ) для машинного обучения, которая использует схему градиентного бустинга.
2019-08-22 11:34:56 +00:00
С помощью этой инструкции вы научитесь применять предобученные модели в ClickHouse: в результате вы запустите вывод модели из SQL.
2019-08-16 06:54:18 +00:00
Чтобы применить модель CatBoost в ClickHouse:
2020-03-20 18:20:59 +00:00
1. [Создайте таблицу ](#create-table ).
2. [Вставьте данные в таблицу ](#insert-data-to-table ).
3. [Интегрируйте CatBoost в ClickHouse ](#integrate-catboost-into-clickhouse ) (Опциональный шаг).
4. [Запустите вывод модели из SQL ](#run-model-inference ).
2019-08-22 05:45:45 +00:00
Подробнее о б обучении моделей в CatBoost, см. [Обучение и применение моделей ](https://catboost.ai/docs/features/training.html#training ).
2019-08-16 06:54:18 +00:00
2021-06-04 20:07:28 +00:00
Вы можете перегрузить модели CatBoost, если их конфигурация была обновлена, без перезагрузки сервера. Для этого используйте системные запросы [RELOAD MODEL ](../sql-reference/statements/system.md#query_language-system-reload-model ) и [RELOAD MODELS ](../sql-reference/statements/system.md#query_language-system-reload-models ).
2019-08-22 11:34:56 +00:00
## Перед началом работы {#prerequisites}
2019-08-16 06:54:18 +00:00
Если у вас еще нет [Docker ](https://docs.docker.com/install/ ), установите е г о .
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2019-08-22 11:34:56 +00:00
[Docker ](https://www.docker.com ) – это программная платформа для создания контейнеров, которые изолируют установку CatBoost и ClickHouse от остальной части системы.
2022-04-09 13:29:05 +00:00
:::
2019-08-23 10:55:34 +00:00
Перед применением модели CatBoost:
2019-08-16 06:54:18 +00:00
**1.** Скачайте [Docker-образ ](https://hub.docker.com/r/yandex/tutorial-catboost-clickhouse ) из реестра:
2020-03-20 18:20:59 +00:00
``` bash
2019-08-16 06:54:18 +00:00
$ docker pull yandex/tutorial-catboost-clickhouse
```
2019-08-22 11:34:56 +00:00
Данный Docker-образ содержит все необходимое для запуска CatBoost и ClickHouse: код, среду выполнения, библиотеки, переменные окружения и файлы конфигурации.
2019-08-16 06:54:18 +00:00
2019-08-22 11:34:56 +00:00
**2.** Проверьте, что Docker-образ успешно скачался:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` bash
2019-08-16 06:54:18 +00:00
$ docker image ls
REPOSITORY TAG IMAGE ID CREATED SIZE
2019-10-10 12:17:39 +00:00
yandex/tutorial-catboost-clickhouse latest 622e4d17945b 22 hours ago 1.37GB
2019-08-16 06:54:18 +00:00
```
2019-08-22 11:34:56 +00:00
**3.** Запустите Docker-контейнер основанный на данном образе:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` bash
2019-08-16 06:54:18 +00:00
$ docker run -it -p 8888:8888 yandex/tutorial-catboost-clickhouse
```
2019-08-22 05:45:45 +00:00
## 1. Создайте таблицу {#create-table}
2019-08-16 06:54:18 +00:00
2019-10-10 12:17:39 +00:00
Чтобы создать таблицу для обучающей выборки:
2019-08-16 06:54:18 +00:00
2019-10-10 12:17:39 +00:00
**1.** Запустите клиент ClickHouse:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` bash
2019-08-16 06:54:18 +00:00
$ clickhouse client
```
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2019-10-10 12:17:39 +00:00
Сервер ClickHouse уже запущен внутри Docker-контейнера.
2022-04-09 13:29:05 +00:00
:::
2019-08-16 06:54:18 +00:00
**2.** Создайте таблицу в ClickHouse с помощью следующей команды:
2020-03-20 18:20:59 +00:00
``` sql
2019-08-16 06:54:18 +00:00
:) CREATE TABLE amazon_train
(
2020-03-20 18:20:59 +00:00
date Date MATERIALIZED today(),
ACTION UInt8,
RESOURCE UInt32,
MGR_ID UInt32,
ROLE_ROLLUP_1 UInt32,
ROLE_ROLLUP_2 UInt32,
ROLE_DEPTNAME UInt32,
ROLE_TITLE UInt32,
ROLE_FAMILY_DESC UInt32,
ROLE_FAMILY UInt32,
2019-08-16 06:54:18 +00:00
ROLE_CODE UInt32
)
2020-01-09 16:42:37 +00:00
ENGINE = MergeTree ORDER BY date
2019-08-16 06:54:18 +00:00
```
2019-10-11 15:36:25 +00:00
**3.** Выйдите из клиента ClickHouse:
2020-03-20 18:20:59 +00:00
``` sql
2019-10-11 15:36:25 +00:00
:) exit
```
2019-10-10 12:17:39 +00:00
## 2. Вставьте данные в таблицу {#insert-data-to-table}
2019-08-16 06:54:18 +00:00
Чтобы вставить данные:
2019-10-11 15:36:25 +00:00
**1.** Выполните следующую команду:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` bash
2019-10-11 15:36:25 +00:00
$ clickhouse client --host 127.0.0.1 --query 'INSERT INTO amazon_train FORMAT CSVWithNames' < ~/amazon/train.csv
2019-08-16 06:54:18 +00:00
```
2019-10-11 15:36:25 +00:00
**2.** Запустите клиент ClickHouse:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` bash
2019-10-11 15:36:25 +00:00
$ clickhouse client
2019-08-16 06:54:18 +00:00
```
2019-10-14 12:39:42 +00:00
**3.** Проверьте, что данные успешно загрузились:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` sql
2019-08-16 06:54:18 +00:00
:) SELECT count() FROM amazon_train
2019-08-22 15:13:36 +00:00
2019-08-16 06:54:18 +00:00
SELECT count()
FROM amazon_train
+-count()-+
2019-10-10 12:17:39 +00:00
| 65538 |
2019-08-16 06:54:18 +00:00
+---------+
```
2019-10-14 10:33:39 +00:00
## 3. Интегрируйте CatBoost в ClickHouse {#integrate-catboost-into-clickhouse}
2019-10-11 15:36:25 +00:00
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2020-03-20 18:20:59 +00:00
**Опциональный шаг.** Docker-образ содержит все необходимое для запуска CatBoost и ClickHouse.
2022-04-09 13:29:05 +00:00
:::
2019-10-14 10:33:39 +00:00
Чтобы интегрировать CatBoost в ClickHouse:
2019-10-11 15:36:25 +00:00
2019-10-14 12:39:42 +00:00
**1.** Создайте библиотеку для оценки модели.
2019-08-16 06:54:18 +00:00
2019-10-14 12:39:42 +00:00
Наиболее быстрый способ оценить модель CatBoost — это скомпилировать библиотеку `libcatboostmodel.<so|dll|dylib>` . Подробнее о том, как скомпилировать библиотеку, читайте в [документации CatBoost ](https://catboost.ai/docs/concepts/c-plus-plus-api_dynamic-c-pluplus-wrapper.html ).
2019-10-14 10:33:39 +00:00
2019-10-14 13:08:46 +00:00
**2.** Создайте в любом месте новую директорию с произвольным названием, например `data` и поместите в нее созданную библиотеку. Docker-образ уже содержит библиотеку `data/libcatboostmodel.so` .
2019-10-14 10:33:39 +00:00
**3.** Создайте в любом месте новую директорию для конфигурации модели с произвольным названием, например `models` .
**4.** Создайте файл конфигурации модели с произвольным названием, например `models/amazon_model.xml` .
**5.** Опишите конфигурацию модели:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` xml
2019-08-16 06:54:18 +00:00
< models >
< model >
<!-- Тип модели. В настоящий момент ClickHouse предоставляет только модель catboost. -->
< type > catboost< / type >
<!-- Имя модели. -->
< name > amazon< / name >
<!-- Путь к обученной модели. -->
< path > /home/catboost/tutorial/catboost_model.bin< / path >
<!-- Интервал обновления. -->
< lifetime > 0< / lifetime >
< / model >
< / models >
```
2019-10-14 13:13:26 +00:00
**6.** Добавьте в конфигурацию ClickHouse путь к CatBoost и конфигурации модели:
2019-10-11 15:36:25 +00:00
2020-03-20 18:20:59 +00:00
``` xml
2019-10-14 10:33:39 +00:00
<!-- Файл etc/clickhouse - server/config.d/models_config.xml. -->
2019-10-14 13:08:46 +00:00
< catboost_dynamic_library_path > /home/catboost/data/libcatboostmodel.so< / catboost_dynamic_library_path >
2019-08-16 06:54:18 +00:00
< models_config > /home/catboost/models/*_model.xml< / models_config >
```
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2021-04-19 17:01:58 +00:00
Вы можете позднее изменить путь к конфигурации модели CatBoost без перезагрузки сервера.
2022-04-09 13:29:05 +00:00
:::
2019-10-10 12:17:39 +00:00
## 4. Запустите вывод модели из SQL {#run-model-inference}
2019-08-16 06:54:18 +00:00
2019-10-10 12:17:39 +00:00
Для тестирования модели запустите клиент ClickHouse `$ clickhouse client` .
2019-08-16 06:54:18 +00:00
2019-08-22 11:34:56 +00:00
Проверьте, что модель работает:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` sql
:) SELECT
modelEvaluate('amazon',
2019-08-16 06:54:18 +00:00
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
2020-03-20 18:20:59 +00:00
ROLE_CODE) > 0 AS prediction,
2019-08-16 06:54:18 +00:00
ACTION AS target
FROM amazon_train
LIMIT 10
```
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2020-04-30 18:19:18 +00:00
Функция [modelEvaluate ](../sql-reference/functions/other-functions.md#function-modelevaluate ) возвращает кортежи (tuple) с исходными прогнозами по классам для моделей с несколькими классами.
2022-04-09 13:29:05 +00:00
:::
2019-08-22 11:34:56 +00:00
Спрогнозируйте вероятность:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` sql
:) SELECT
modelEvaluate('amazon',
2019-08-16 06:54:18 +00:00
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
ROLE_CODE) AS prediction,
2020-03-20 18:20:59 +00:00
1. / (1 + exp(-prediction)) AS probability,
2019-08-16 06:54:18 +00:00
ACTION AS target
FROM amazon_train
LIMIT 10
```
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2020-04-30 18:19:18 +00:00
Подробнее про функцию [exp() ](../sql-reference/functions/math-functions.md ).
2022-04-09 13:29:05 +00:00
:::
2019-08-22 11:34:56 +00:00
Посчитайте логистическую функцию потерь (LogLoss) на всей выборке:
2019-08-16 06:54:18 +00:00
2020-03-20 18:20:59 +00:00
``` sql
2019-08-16 06:54:18 +00:00
:) SELECT -avg(tg * log(prob) + (1 - tg) * log(1 - prob)) AS logloss
2020-03-20 18:20:59 +00:00
FROM
2019-08-16 06:54:18 +00:00
(
2020-03-20 18:20:59 +00:00
SELECT
modelEvaluate('amazon',
2019-08-16 06:54:18 +00:00
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
ROLE_CODE) AS prediction,
2020-03-20 18:20:59 +00:00
1. / (1. + exp(-prediction)) AS prob,
2019-08-16 06:54:18 +00:00
ACTION AS tg
FROM amazon_train
)
2019-08-22 11:34:56 +00:00
```
2022-04-09 13:29:05 +00:00
:::note "Примечание"
2020-07-29 06:49:57 +00:00
Подробнее про функции [avg() ](../sql-reference/aggregate-functions/reference/avg.md#agg_function-avg ), [log() ](../sql-reference/functions/math-functions.md ).
2022-04-09 13:29:05 +00:00
:::