|
|
|
@ -16,6 +16,7 @@ import pymysql
|
|
|
|
|
|
|
|
|
|
# ==================== 数据库 ====================
|
|
|
|
|
# 数据库配置(需根据实际环境修改)
|
|
|
|
|
input_table_name = "only_text"
|
|
|
|
|
DB_CONFIG = {
|
|
|
|
|
'host': 'localhost',
|
|
|
|
|
'user': 'root',
|
|
|
|
@ -210,8 +211,8 @@ def main():
|
|
|
|
|
#input_data = read_csv_to_array(r'./output_lite.csv')
|
|
|
|
|
|
|
|
|
|
# 预处理
|
|
|
|
|
id_data = read_from_table("only_text", "id")
|
|
|
|
|
text_data = read_from_table("only_text", "text")
|
|
|
|
|
id_data = read_from_table(input_table_name, "id")
|
|
|
|
|
text_data = read_from_table(input_table_name, "text")
|
|
|
|
|
data = [[id_data[i],text_data[i]] for i in range(len(id_data))]
|
|
|
|
|
#print(raw_text) #把id去掉,只返回text字段
|
|
|
|
|
# 初始化提取器
|
|
|
|
|