update

contropist · Oct 16, 2023 · f00f158 · f00f158
1 parent d3acd57
commit f00f158
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/code/finglm_all/prepare_model/text2es.py b/code/finglm_all/prepare_model/text2es.py
@@ -180,6 +180,7 @@ def prepare_es(self, text, re_dict):
             body['query']={}
             body['query']['bool']= {}
             body['query']['bool']['must'] = list1
+            body['_source'] = ['公司名称']
             body['sort'] = [dict2]
             body['size'] = num
             es = text + '####' + str(body)
@@ -219,6 +220,7 @@ def prepare_es(self, text, re_dict):
             body['query']={}
             body['query']['bool']= {}
             body['query']['bool']['must'] = list1
+            body['_source'] = ['公司名称']
             body['sort'] = [dict2]
             body['size'] = num
             es = text + '####' + str(body)
@@ -246,6 +248,7 @@ def prepare_es(self, text, re_dict):
             body['query']={}
             body['query']['bool']= {}
             body['query']['bool']['must'] = list1
+            body['_source'] = ['公司名称']
             body['sort'] = [dict2]
             body['size'] = 1
             es = text + '####' + str(body)