|
|
@@ -428,6 +428,7 @@ def tableToText(soup):
|
|
|
return inner_table,head_list
|
|
|
|
|
|
def set_head_model(inner_table):
|
|
|
+ origin_inner_table = copy.deepcopy(inner_table)
|
|
|
for i in range(len(inner_table)):
|
|
|
for j in range(len(inner_table[i])):
|
|
|
# 删掉单格前后符号，以免影响表头预测
|
|
|
@@ -442,7 +443,7 @@ def tableToText(soup):
|
|
|
# 组合结果
|
|
|
for i in range(len(inner_table)):
|
|
|
for j in range(len(inner_table[i])):
|
|
|
- inner_table[i][j] = [inner_table[i][j], int(predict_list[i][j])]
|
|
|
+ inner_table[i][j] = [origin_inner_table[i][j][0], int(predict_list[i][j])]
|
|
|
head_list = sliceTable(inner_table)
|
|
|
return inner_table, head_list
|
|
|
|
