连接数据库、处理。要求数据库中存在only_text表(可自行修改),即只包含id和text的处理后表格。

main
p5rqvkz4o 4 months ago
parent 52efd7247e
commit 283d4930a0

@ -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字段
# 初始化提取器

Loading…
Cancel
Save