hm_ir_train_app.py 422 B

12345678910111213
  1. from hnsw_model_train import HNSW_Model_Train
  2. from ir_db_establish import Inverted_Index_Establish
  3. from formula_process import formula_init
  4. if __name__ == "__main__":
  5. # 向量检索模型训练
  6. hm_train = HNSW_Model_Train()
  7. hm_train()
  8. # 初始化倒排索引建立(n-grams为True)
  9. sql_etl = Inverted_Index_Establish(n_grams_flag=True)
  10. sql_etl()
  11. # 公式初始化数据训练
  12. formula_init()