pmkhanh7890's picture
refactor code + fix bug of label after grouping url
00b1038
raw
history blame
424 Bytes
import pandas as pd
class TextDetector:
def __init__(self):
self.input = None
self.prediction_label: list[str] = ["UNKNOWN"]
self.prediction_score: list[float] = [0.0]
self.grouped_url_df: pd.DataFrame = pd.DataFrame()
# For formatting ouput tables
self.ordinary_user_table: list = []
self.fact_checker_table: list = []
self.governor_table: list = []