Udostępnij przez


Używanie biblioteki ai.classify z biblioteką pandas

Funkcja ai.classify używa generowania sztucznej inteligencji do kategoryzowania tekstu wejściowego zgodnie z wybranymi etykietami niestandardowymi z jednym wierszem kodu.

Uwaga / Notatka

Przegląd

Funkcja ai.classify rozszerza klasę serii pandas. Aby przypisać etykiety dostarczone przez użytkownika do każdego wiersza wejściowego, wywołaj funkcję w kolumnie tekstowej ramki danych biblioteki pandas.

Funkcja zwraca serię pandas zawierającą etykiety klasyfikacji, które mogą być przechowywane w nowej kolumnie ramki danych.

Wskazówka

Zalecamy użycie funkcji ai.classify z co najmniej dwiema etykietami wejściowymi.

Składnia

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

Parametry

Name Description
labels
Required
Co najmniej jeden ciąg reprezentujący zestaw etykiet klasyfikacji do dopasowania do wejściowych wartości tekstowych.

Zwraca

Funkcja zwraca serii pandas, która zawiera etykietę klasyfikacji dla każdego wiersza tekstu wejściowego. Jeśli nie można sklasyfikować wartości tekstowej, odpowiednia etykieta jest 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)

Ta przykładowa komórka kodu udostępnia następujące dane wyjściowe:

Zrzut ekranu przedstawiający ramkę danych z kolumnami