Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def max_sentiment_class(row):
- sentiment_columns = ["neg", "neu", "pos"]
- idx_max = np.argmax(row[sentiment_columns].values)
- return sentiment_columns[idx_max]
- df['sentiment_class'] = df.apply(lambda x: max_sentiment_class(x), axis=1)
Add Comment
Please, Sign In to add comment