def analyze_sentiment(review): | |
response = openai.completions.create( | |
model="text-davinci-003", # You can choose the model version | |
prompt=f"Analyze the sentiment of this review: \"{review}\". Is it positive, negative, or neutral?", | |
max_tokens=60 | |
) | |
sentiment = response.choices[0].text.strip() | |
return sentiment | |
df_short["Polarity"] = df_short.apply(lambda x:analyze_sentiment(x["Text"]), axis=1) | |
df_short |