a guest Oct 23rd, 2018 60 Never
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)
RAW Paste Data