read_annotated_meld_data.py

from collections import Counter
from diswiz.utils_server import EDAs
from src.plot_utils import plot_bars_plot
from src.read_annotated_data_utils import read_data

utt_Speaker, utt, utt_Emotion, utt_EDAs, utt_Sentiment = read_data('annotated_data/eda_meld_emotion_dataset.csv',
                                                                   meld_data=True)
colors_emo = ['Green', 'Blue', 'Black', 'Olive', 'Mediumvioletred', 'Orange', 'Red', 'brown']
emotions = ['joy', 'surprise', 'neutral', 'fear', 'sadness', 'disgust', 'anger']
colors_sent = ['Limegreen', 'Black', 'Darkorange', 'White']
sentiments = ['positive', 'neutral', 'negative', 'White']

tags = sorted(list(Counter(utt_EDAs).keys()))

c = Counter(utt_EDAs)
x = [(i, c[i] / len(utt_EDAs) * 100.0) for i, count in c.most_common()]
for item in x:
    print(item[0], round(item[1], 2))

stack_emotions_values = []
stack_da_emotions, stack_da_sentiments = {}, []
pass_emotions, pass_values = [], []
for tag in tags:
    if tag is not str:
        pass
    temp_emotion, temp_sentiment = [], []
    for i in range(len(utt)):
        if str(utt_EDAs[i]) == str(tag):
            temp_emotion.append(utt_Emotion[i])
            temp_sentiment.append(utt_Sentiment[i])
    data_emotion = Counter(temp_emotion)
    data_sentiment = Counter(temp_sentiment)
    values_emotion, values_sentiment = [], []
    for emotion in emotions:
        values_emotion.append(data_emotion[emotion])
    for sentiment in sentiments:
        values_sentiment.append(data_sentiment[sentiment])

    try:
        if tag == 'xx':
            title = tag + '\n' + "Unknown EDA"
        elif tag == 'fo_o_fw_"_by_bc':
            tag = 'fo'
            title = tag + '\n' + EDAs[tag]
        elif tag in ['aap_am', 'arp_nd']:
            pass
        else:
            title = tag + '\n' + EDAs[tag]
    except TypeError:
        title = str(tag)

    pass_emotions = emotions
    pass_values = values_emotion
    pass_values.extend([sum(values_emotion)])

    pass_values_sent = values_sentiment
    pass_values_sent.extend([sum(values_sentiment)])
    # plot_normal_bars(emotions, values, title)
    # plot_eda_usage(emotions, pass_values, title, colors_emo, sentiments, pass_values_sent, colors_sent, plot_pie=True)
    stack_da_emotions[title] = values_emotion
    stack_da_sentiments.append(values_sentiment)
    stack_emotions_values.append(values_emotion)

plot_bars_plot(stack_emotions_values, emotions, colors_emo, tags,
               test_show_plot=False, data='meld', type_of='emotion', plot_sankeys=True)  # , save_eps=True)

plot_bars_plot(stack_da_sentiments, sentiments, colors_sent, tags,
               test_show_plot=False, data='meld', type_of='sentiment')  # , save_eps=True)

print('ran read_annotated_meld_data.py')