Поделиться через


Использование ai.classify с pandas

Функция ai.classify использует генерированный ИИ для классификации входного текста в соответствии с настраиваемыми метками, которые вы выбрали, с одной строкой кода.

Замечание

Обзор

Функция ai.classify расширяет класс серии pandas. Чтобы назначить предоставленные пользователем метки каждой входной строке, вызовите функцию в текстовом столбце кадра данных Pandas.

Функция возвращает объект pandas Series, содержащий метки классификации, которые можно хранить в новом столбце DataFrame.

Подсказка

Рекомендуется использовать функцию ai.classify с по крайней мере двумя входными метками.

Синтаксис

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

Параметры

Имя Description
labels
Обязательно
Одна или несколько строк , представляющих набор меток классификации для сопоставления с входными текстовыми значениями.

Возвраты

Функция возвращает pandas Series, содержащая метку классификации для каждой входной текстовой строки. Если текстовое значение не может быть классифицировано, ему присваивается метка 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)

В этом примере ячейка кода предоставляет следующие выходные данные:

Снимок экрана: кадр данных с столбцами descriptions и category. В столбце