# this is the entrance of the auto-ER procedure
from md_discovery.md_discover import md_discover
from ml_er.ml_entity_resolver import ml_er
from hpo.er_model_hpo import ml_er_hpo
from settings import *


def run(rounds: int):
    hp_config = None
    iter_round = 1
    for i in range(0, rounds):
        ml_er(iter_round, hp_config)
        md_discover()
        hp_config = ml_er_hpo()
        iter_round += 1
    ml_er(iter_round, hp_config)
    return


if __name__ == '__main__':
    # todo
    #  使用drop删除特征向量中的列？(如删除id相关特征)
    run(1)  # 迭代3轮
    # ml_er(1)
    # todo 将优化结果与参数输出到文件中
    #  通过ml_entity_resolver.ml_er()输出,同时输出参数配置信息
    print(ltable_path)