10,1 тыс подписчиков

🐼Анализ данных на основе промптов с помощью PandasAI

Pandas AI — это библиотека Python со встроенным генеративным искусственным интеллектом или языковой моделью. Она превращает ваши фреймы данных в собеседников. Больше никаких бесконечных часов разглядывания строк и столбцов.

PandasAI может генерировать визуализации на основе промптов, хотя PandasAI не является заменой Pandas, это дополняющая библиотека, которая предоставляет супервозможности ИИ для работы с данными.

🔵 Попробуем PandasAI на наборе данных Spotify Kaggle:

# Import Dependencies

import pandas as pd

import opendatasets as od

import pandasai as pai

from pandasai.llm.openai import OpenAI

# Get Spotify Data from kaggle

od.download("https://www.kaggle.com/datasets/amitanshjoshi/spotify-1million-tracks")

spotify_data = '<location>/spotify_data.csv'

# Read Spotify Dataset

df = pd.read_csv(spotify_data)

# Initiate OpenAI LLM model for spotify dataset.

llm = OpenAI("<OpenAI API Key>")

pandas_ai = pai.PandasAI(llm)

🔵 Теперь мы готовы использовать наш кадр данных панды, управляемый OpenAI.

# Let us get top 10 artist in year 2012 via a prompt

pandas_ai(df, prompt='Which are top 10 artists in 2012?')p

▪ Результат

Пробовали данную библиотеку?

@machinelearning

1 минута

13 июля 2023

653 читали