INNER CODE UNIT · Python
get_embeddings
business-science/free-ai-tips · 002_customer_churn_ai_ml/customer_churn_ai_ml.py:79
def get_embeddings(text):
response = client.embeddings.create(
input=[text],
model=EMBEDDING_MODEL
)
# Access data as attributes instead of dict indexing:
embedding = response.data[0].embedding
return embedding
df['summary_embedding'] = df['ticket_summary'].apply(get_embeddings)
# df.to_csv(PATH_ROOT + "/data/customer_churn_summary_embeddings.csv", index=False)
df = pd.read_csv(PATH_ROOT + "/data/customer_churn_summary_embeddings.csv")
df
# ---------------------------