From 2b5d801826e77f059aff801852e026efc6ec566b Mon Sep 17 00:00:00 2001 From: ThijsVroegh Date: Mon, 9 Dec 2024 09:21:39 +0100 Subject: [PATCH] Added frequencies of future verbs in action stats --- orangecontrib/storynavigation/modules/actionanalysis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orangecontrib/storynavigation/modules/actionanalysis.py b/orangecontrib/storynavigation/modules/actionanalysis.py index dcd46d5..dfed512 100644 --- a/orangecontrib/storynavigation/modules/actionanalysis.py +++ b/orangecontrib/storynavigation/modules/actionanalysis.py @@ -373,7 +373,7 @@ def generate_action_analysis_results(self, story_elements_df, callback=None): if word_col is None: word_col = 'token_text_lowercase' - past_or_present_tense_verbs_df = story_elements_df[story_elements_df['story_navigator_tag'].isin(['PRES_VB', 'PAST_VB'])] + past_or_present_tense_verbs_df = story_elements_df[story_elements_df['story_navigator_tag'].isin(['PRES_VB', 'PAST_VB','FUTURE_VB'])] result_df = past_or_present_tense_verbs_df.groupby(['storyid', 'segment_id', 'story_navigator_tag'])[word_col].agg(word_col='nunique').reset_index().rename(columns={word_col: "tense_freq"}) return result_df