Spaces:
Runtime error
Runtime error
removed unnecessary methods
Browse files
textclassifier/TextClassifier.py
CHANGED
@@ -324,26 +324,6 @@ class TextClassifier:
|
|
324 |
print("Created {}.".format(filename))
|
325 |
return None
|
326 |
|
327 |
-
|
328 |
-
@staticmethod
|
329 |
-
def split_topics(text:str):
|
330 |
-
"""
|
331 |
-
splits a string of format '1. topic1 2. topic2 3. topic3' into a list of topics
|
332 |
-
:param text: string of format '1. topic1 2. topic2 3. topic3'
|
333 |
-
:return: list of topics
|
334 |
-
"""
|
335 |
-
topics = text.split(' ')
|
336 |
-
topics = [topic.split('.')[1] for topic in topics if int(topic.split('.')[0]) > 0]
|
337 |
-
return topics
|
338 |
-
|
339 |
-
def plot(self, type: int):
|
340 |
-
"""
|
341 |
-
Plots the sentiment of the tweets. Since self.df['topics'] is in the format '1. topic 1 2. topic 2 3. topic 3',
|
342 |
-
we first split the string into new columns
|
343 |
-
:param type:
|
344 |
-
:return: plot
|
345 |
-
"""
|
346 |
-
|
347 |
def get_dataframe(self):
|
348 |
"""
|
349 |
Returns the dataframe.
|
|
|
324 |
print("Created {}.".format(filename))
|
325 |
return None
|
326 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
327 |
def get_dataframe(self):
|
328 |
"""
|
329 |
Returns the dataframe.
|