Upload sentiment_labeling.py

sentiment_labeling.py

@@ -0,0 +1,20 @@
+import pandas as pd
+
+def label_sentiment(rating):
+    """Label sentiment based on the rating."""
+    if rating in [1, 2]:
+        return 'negative'
+    elif rating == 3:
+        return 'neutral'
+    elif rating in [4, 5]:
+        return 'positive'
+    else:
+        return 'unknown'
+
+def add_sentiment_column(data: pd.DataFrame, rating_col_name='rating'):
+    """Add a sentiment column to the dataframe based on the ratings."""
+    if rating_col_name not in data.columns:
+        raise ValueError(f"Column '{rating_col_name}' not found in the dataframe.")
+    
+    data['sentiment'] = data[rating_col_name].apply(label_sentiment)
+    return data