You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
matching_dependency/entrance.py

28 lines
763 B

1 year ago
# this is the entrance of the auto-ER procedure
1 year ago
from md_discovery.md_discover import md_discover
1 year ago
from ml_er.ml_entity_resolver import ml_er
from hpo.er_model_hpo import ml_er_hpo
from settings import *
1 year ago
1 year ago
def run(rounds: int):
hp_config = None
iter_round = 1
for i in range(0, rounds):
ml_er(iter_round, hp_config)
1 year ago
md_discover()
1 year ago
hp_config = ml_er_hpo()
iter_round += 1
ml_er(iter_round, hp_config)
1 year ago
return
if __name__ == '__main__':
1 year ago
# todo
# 使用drop删除特征向量中的列(如删除id相关特征)
run(1) # 迭代3轮
1 year ago
# ml_er(1)
1 year ago
# todo 将优化结果与参数输出到文件中
# 通过ml_entity_resolver.ml_er()输出,同时输出参数配置信息
1 year ago
print(ltable_path)