From 7c401c75ac0374197fed895d3104e1bff8893b0b Mon Sep 17 00:00:00 2001 From: "Anna Jung (VMware)" Date: Wed, 1 Dec 2021 14:09:57 -0600 Subject: [PATCH] Fix missing Number column in the annotated dataset Signed-off-by: Anna Jung (VMware) --- ml-conversational-analytic-tool/featureVector.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ml-conversational-analytic-tool/featureVector.py b/ml-conversational-analytic-tool/featureVector.py index e1b0af7..a09a723 100644 --- a/ml-conversational-analytic-tool/featureVector.py +++ b/ml-conversational-analytic-tool/featureVector.py @@ -163,6 +163,7 @@ def pullStringConversation(self, export_filename="", export=True): # Store each interaction and pull URL for export string_conversations = [] pull_urls = [] + pull_numbers = [] for index, row in self.raw_data.iterrows(): # Make pull message @@ -181,9 +182,11 @@ def pullStringConversation(self, export_filename="", export=True): comment_row["Body"]) string_conversations.append(conversation.encode("ascii", "ignore").decode()) pull_urls.append(row["URL"]) + pull_numbers.append(row["Number"]) # Export converation field dataset export_df = pd.DataFrame() + export_df["Number"] = pull_numbers export_df["URL"] = pull_urls export_df["Thread"] = string_conversations