Модели классификаторов на базе YandexGPT

Сервис Yandex AI Studio предоставляет доступ к моделям классификаторов на базе YandexGPT Lite и YandexGPT Pro, которые позволяют классифицировать передаваемые в промптах текстовые запросы. Классификация реализована в AI Studio Text Classification API.

Если стандартных моделей вам недостаточно, вы можете дообучить классификатор на базе YandexGPT Lite, чтобы модель точнее классифицировала ваши запросы. Чтобы обратиться к дообученной модели классификатора, используйте метод classify Text Classification API.

Модель URI Режимы работы
Классификатор по промпту на базе YandexGPT Lite cls://<идентификатор_каталога>/yandexgpt-lite/latest Синхронный
Классификатор по промпту на базе YandexGPT Pro cls://<идентификатор_каталога>/yandexgpt/latest Синхронный
Дообученный классификатор cls://<URI_базовой_модели>/<версия>@<суффикс_дообучения> Синхронный

Обращение к моделям

Вы можете обращаться к моделям классификаторов несколькими способами.

При работе с моделями классификаторов через Yandex AI Studio SDK используйте один из следующих форматов:

  • Название модели, передается в виде строки.

    model = (
          sdk.models.text_classifiers("yandexgpt-lite")
        )

  • Название и версия модели, передаются в виде строк в полях model_name и model_version соответственно.

    model = (
          sdk.models.text_classifiers(model_name="yandexgpt", model_version="latest")
        )

  • URI модели, передается в виде строки, содержащей полный URI модели. Также используйте этот способ для обращения к дообученным моделям.

    model = (
          sdk.models.text_classifiers("cls://b1gt6g8ht345********/yandexgpt/latest")
        )

Чтобы обратиться к модели через REST API или gRPC API, в поле modelUri тела запроса укажите URI модели, содержащий идентификатор каталога. Сегмент /latest указывает на версию модели и является необязательным. Чтобы обратиться к модели классификатора на базе YandexGPT, используйте метод/вызов fewShotClassify Text Classification API.

Пример:

{
          "modelUri": "cls://b1gt6g8ht345********/yandexgpt-lite/latest"
          ...
        }

Для обращения к Latest версии необязательно задавать версию модели явно, поскольку версия Latest используется по умолчанию.

См. также

Предыдущая
Поддерживаемые виды классификации
Следующая
Эмбеддинги