storytelling / src /story_gen_test.py
jitesh's picture
adds probability_emote page
441ec1f
# %%
import printj
from story_gen import StoryGenerator
gen = StoryGenerator()
# # %%
# story_till_now, emotion = gen.story(story_till_now='Hello, I\'m a language model,', num_generation=3, length=10)
# printj.purple(story_till_now)
# printj.yellow(emotion)
# %%
gen.get_stats(story_till_now="For myriad of eons i’ve forgotten who I really was, harvesting the essence of all existence.",
length=10, num_generation=3, num_tests=50)
# %%
gen.save_stats('/home/jitesh/haru/ist/results/a.xlsx')
# %%
data=gen.stats_df[gen.stats_df.sentence_no==3]
import seaborn as sns
sns.set_theme(style="whitegrid")
# ax = sns.violinplot(x="day", y="total_bill", data=tips)
ax = sns.violinplot(x="reaction_weight", y="num_reactions", data=data).set_title('Analysing ProbabilityEmote (Max reactions=3)')
# %%
gen.stats_df[gen.stats_df.sentence_no==3]
# %%
import re
len(re.findall(r'\w+', 'line ive '))
# %%