次の方法で共有

ai_classify 関数

  • [アーティクル]

適用対象:check marked yes Databricks SQL

重要

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

プレビューでは、

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

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

要件

重要

現時点で使用できる基になるモデルは、Apache 2.0 ライセンスまたは Llama 2 コミュニティ ライセンスの下でライセンスされています。 Databricks では、該当する使用条件に準拠するために、これらのライセンスを確認することを推奨しています。 将来、Databricks の内部ベンチマークの結果でより良いパフォーマンスを示すモデルが現れた場合、Databricks はモデル (およびこのページで提供される該当ライセンスの一覧) を変更する可能性があります。

現在のところ、Mixtral-8x7B Instruct がこれらの 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