tujintao преди 10 месеца
родител
ревизия
583dce2118
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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)