tujintao 10 months ago
parent
commit
583dce2118
1 changed files with 1 additions and 1 deletions
  1. 1 1
      db_train_app.py

+ 1 - 1
db_train_app.py

@@ -35,7 +35,7 @@ def clear_embedding_train(mongo_coll, mongo_find_dict, sup, sub):
 # 知识点转换成id用于mongodb检索/计算物理量/计算求解类型
 def convert_knowledge2id(mongo_coll, mongo_find_dict, sup, sub):
     # 加载知识点转ID数据
-    with open("model_data/keyword_mapping.json", 'r', encoding="utf8") as f:
+    with open(config.keyword_mapping_path, 'r', encoding="utf8") as f:
         knowledge2id = json.load(f)["knowledge2id"]
     dim_classify = Dimension_Classification(dim_mode=0)
     origin_dataset = mongo_coll.find(mongo_find_dict, no_cursor_timeout=True, batch_size=5)