Sentiment_Analysis/data_analysis.py at master · Elie60/Sentiment_Analysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
from clean_tweets_dataframe import Clean_Tweets
import os
import sys
import inspect
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
from wordcloud import STOPWORDS, WordCloud
from gensim import corpora
import numpy as np
import string

def run():
    processed_tweets = pd.read_csv('./data/processed_tweet_data.csv')

    print("The number of missing value(s) based on columns:\n{}".format(
        processed_tweets.isnull().sum()))
    print("The number of missing value(s): {}".format(
        processed_tweets.isnull().sum().sum()))
    print("Columns having missing value(s):{}".format(
        processed_tweets.columns[processed_tweets.isnull().any()]))

    currentdir = os.path.dirname(os.path.abspath(
        inspect.getfile(inspect.currentframe())))
    parentdir = os.path.dirname(currentdir)
    sys.path.insert(0, parentdir)

    tweets_df = pd.DataFrame(
        columns=['original_text', 'sentiment', 'lang', 'hashtags'])

    tweets_df['original_text'] = processed_tweets['original_text'].to_list()
    tweets_df['sentiment'] = processed_tweets['sentiment'].to_list()
    tweets_df['lang'] = processed_tweets['lang'].to_list()
    tweets_df['hashtags'] = processed_tweets['hashtags'].to_list()

    tweet_cleaner = Clean_Tweets()
    tweet_cleaner.add_clean_text(tweets_df)

    tweets_df = tweets_df.fillna("")

    tweets_df = tweet_cleaner.remove_non_english_tweets(tweets_df)

    tweets_df['clean_text'] = tweets_df['clean_text'].astype(str)
    tweets_df['clean_text'] = tweets_df['clean_text'].apply(lambda x: x.lower())
    tweets_df['clean_text'] = tweets_df['clean_text'].apply(
        lambda x: x.translate(str.maketrans(' ', ' ', string.punctuation)))
    tweets_df['sentiment'] = tweets_df['sentiment'].astype(int)
    tweets_df['hashtags'] = tweets_df['hashtags'].astype(str)
    tweets_df['hashtags'] = tweets_df['hashtags'].apply(lambda x: x.lower())
    tweets_df['hashtags'] = tweets_df['hashtags'].apply(
        lambda x: x.translate(str.maketrans(' ', ' ', string.punctuation)))

    flattened_words_df = pd.DataFrame(
        [word for words_list in tweets_df.clean_text
        for word in words_list.split(' ')],
        columns=['word'])

    flattened_hashtags_df = pd.DataFrame(
        [hashtag for hashtags_list in tweets_df.hashtags
        for hashtag in hashtags_list.split(',')],
        columns=['hashtag'])

    flattened_hashtags_df.drop(
        flattened_hashtags_df[flattened_hashtags_df['hashtag'] == ""].index, inplace=True)
    flattened_hashtags_df.reset_index(drop=True, inplace=True)

    tweets_df = tweets_df.drop(['original_text'], axis=1)

    tweets_df.to_csv('./data/model_ready_data.csv', index=False)
    print('Model Ready Data Successfully Saved.!!!')

if __name__ == '__main__':
    run()