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.

13 lines
475 B

from sentence_transformers import SentenceTransformer
directory_path = '/root/hjt/DeepMatcher Dataset/Structured/Amazon-Google'
er_output_dir = '/root/hjt/md_bayesian_er_ditto/ml_er/output'
md_output_dir = '/root/hjt/md_bayesian_er_ditto/md_discovery/output'
hpo_output_dir = '/root/hjt/md_bayesian_er_ditto/hpo/output'
# model = SentenceTransformer('/root/hjt/all-MiniLM-L6-v2')
interpre_weight = 0 # 可解释性权重
support_threshold = 1
confidence_threshold = 0.75