def main(): # 数据加载 date = sys.argv[1] path_prefix = "./new_data" df = pd.read_pickle( os.path.join(path_prefix, "%s_select_comments.p" % date)) # 模型加载 s = Sentiment() df = s.sentiment_detect(df, on=["content"], srcs=["province"], dists=["region_1", "region_2", "region_3"]) df_freq = s.table_record() # 结果保存 df.to_pickle(os.path.join(path_prefix, "%s_sentiment.p" % date)) df_freq.to_pickle(os.path.join(path_prefix, "%s_senti_freq.p" % date)) print(df) print(df_freq)