File size: 508 Bytes
e2b2661
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import pandas as pd
from sentiment import compute_sentiment

def process_stream(path="data/streaming/new_tweets.csv"):
    df = pd.read_csv(path)

    df["sentiment"] = df["text"].apply(compute_sentiment)

    # Aggregate daily sentiment
    daily = (
        df.groupby(["date", "ticker"])["sentiment"]
        .mean()
        .reset_index()
    )

    daily.to_csv("data/processed/new_sentiment.csv", index=False)
    print("Processed new sentiment data")


if __name__ == "__main__":
    process_stream()