from duel import *


def do_duel():
    set_random_seed(20200619)

    # 训练实体链接模型
    train_entity_linking_model('EL_BASE_EPOCH0.ckpt')
    generate_link_tsv_result('EL_BASE_EPOCH0.ckpt')

    # 训练实体分类模型
    train_entity_typing_model('ET_BASE_EPOCH1.ckpt')
    generate_type_tsv_result('ET_BASE_EPOCH1.ckpt')

    # 结果查看
    el_ret = pd.read_csv(DATA_PATH + "result/ET_TEST_RESULT.tsv", sep='\t')
    print(el_ret.head())


if __name__ == '__main__':
    do_duel()