次の方法で共有

ai_classify 関数

  • [アーティクル]

適用対象: Databricks SQL

重要

この機能はパブリック プレビュー段階にあります。

このプレビューでは:

  • 基になる言語モデルでは複数の言語を処理できますが、これらの関数は英語用にチューニングされています。
  • 基になる Foundation Model API にはレート制限があります。 これらの制限を更新するには、「Foundation Model API の制限」を参照してください。

ai_classify() 関数を使用すると、最先端の生成 AI モデルを呼び出し、SQL を使用して指定したラベルに従って入力テキストを分類できます。 この関数は、Databricks Foundation Model API によって利用できるチャット モデル提供エンドポイントを使用します。

要件

重要

現時点で使用できる基になるモデルは、Apache 2.0 ライセンスまたは Llama 2 Community License に基づいてライセンス付与されます。 Databricks では、該当する使用条件に準拠するために、これらのライセンスを確認することをお勧めしています。 将来、Databricks の内部ベンチマークに従ってパフォーマンスが向上するモデルが開発された場合、Databricks によってモデルが変更される可能性があります (変更に伴い、このページで提供されている該当するライセンスの一覧も変更されます)。

現在、Mixtral-8x7B 指示は、これらの AI 関数を実行する基になるモデルとなっています。

構文

ai_classify(content, labels)

引数

  • content: STRING 式。分類するテキスト。
  • labels: ARRAY<STRING> リテラル。期待される出力の分類ラベル。 2 個以上 20 個以下の要素を含む必要があります。

返品

STRING です。 この値は、labels 引数で指定された文字列のいずれかとマッチします。 content が分類できない場合は null を返します。

例

> SELECT ai_classify("My password is leaked.", ARRAY("urgent", "not urgent"));
  urgent

> SELECT
    description,
    ai_classify(description, ARRAY('clothing', 'shoes', 'accessories', 'furniture')) AS category
  FROM
    products