Compartilhar via

Usar ai.classify com pandas

A ai.classify função usa IA generativa para categorizar o texto de entrada de acordo com os rótulos personalizados escolhidos, com uma única linha de código.

Observação

Visão geral

A função ai.classify estende a classe pandas Series. Para atribuir rótulos fornecidos pelo usuário a cada linha de entrada, chame a função em uma coluna de texto de um DataFrame pandas.

A função retorna uma pandas Series que contém rótulos de classificação, que podem ser armazenados em uma nova coluna do DataFrame.

Dica

É recomendável usar a função ai.classify com pelo menos dois rótulos de entrada.

Sintaxe

df["classification"] = df["input"].ai.classify("category1", "category2", "category3")

Parâmetros

Nome Description
labels
Obrigatório		 Uma ou mais cadeias de caracteres que representam o conjunto de rótulos de classificação a serem correspondentes aos valores de texto de entrada.

Devoluções

A função retorna uma pandas Series que contém um rótulo de classificação para cada linha de texto de entrada. Se um valor de texto não puder ser classificado, o rótulo correspondente será null.

Example

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        "This duvet, lovingly hand-crafted from all-natural fabric, is perfect for a good night's sleep.",
        "Tired of friends judging your baking? With these handy-dandy measuring cups, you'll create culinary delights.",
        "Enjoy this *BRAND NEW CAR!* A compact SUV perfect for the professional commuter!"
    ], columns=["descriptions"])

df["category"] = df['descriptions'].ai.classify("kitchen", "bedroom", "garage", "other")
display(df)

Esta célula de código de exemplo fornece a seguinte saída:

Captura de tela de um quadro de dados com colunas

Conteúdo relacionado

  • Last updated on