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.
24 lines
1.2 KiB
24 lines
1.2 KiB
from sentence_transformers import SentenceTransformer
|
|
import numpy as np
|
|
|
|
ltable_path = r'E:\Data\Research\Projects\matching_dependency\datasets\Abt-Buy\tableA.csv'
|
|
rtable_path = r'E:\Data\Research\Projects\matching_dependency\datasets\Abt-Buy\tableB.csv'
|
|
mapping_path = r'E:\Data\Research\Projects\matching_dependency\datasets\Abt-Buy\matches.csv'
|
|
mapping_lid = 'idAbt' # mapping表中左表id名
|
|
mapping_rid = 'idBuy' # mapping表中右表id名
|
|
ltable_id = 'id' # 左表id字段名称
|
|
rtable_id = 'id' # 右表id字段名称
|
|
target_attr = 'id' # 进行md挖掘时的目标字段
|
|
# lr_attrs_map = {} # 如果两个表中存在对应字段名称不一样的情况，将名称加入列表便于调整一致
|
|
|
|
model = SentenceTransformer('E:\\Data\\Research\\Models\\roberta-large-nli-stsb-mean-tokens')
|
|
similarity_threshold = 0.2
|
|
support_threshold = 100
|
|
confidence_threshold = 0.4
|
|
interpre_weight = 0.5 # 可解释性权重
|
|
|
|
er_output_dir = 'E:\\Data\\Research\\Projects\\matching_dependency\\ml_er\\output\\'
|
|
md_output_dir = 'E:\\Data\\Research\\Projects\\matching_dependency\\md_discovery\\output\\'
|
|
hpo_output_dir = 'E:\\Data\\Research\\Projects\\matching_dependency\\hpo\\output\\'
|
|
|