def select_e2e():
params = {
'model_name': MODEL_NAME,
'diversity_param_tuple': DIVERSITY_PARAM_TUPLE,
'cos_threshold': COS_THRESHOLD,
'rm_dialog': RM_DIALOG,
}
select_sent.select_end2end(**params)
def select_e2e():
ir_rec_dp = join(path_parser.summary_rank, qa_config.IR_RECORDS_DIR_NAME)
params = {
'model_name': qa_config.QA_MODEL_NAME_BERT,
'cos_threshold': 1.0,
'retrieved_dp': ir_rec_dp,
}
select_sent.select_end2end(**params)
def select_e2e():
"""
No redundancy removal is applied here.
"""
params = {
'model_name': MODEL_NAME,
'cos_threshold': COS_THRESHOLD,
}
select_sent.select_end2end(**params)
def select_e2e():
params = {
'model_name': model_name,
'diversity_param_tuple': centrality_config.DIVERSITY_PARAM_TUPLE,
'cos_threshold': centrality_config.
COS_THRESHOLD, # do not pos cosine similarity criterion?
'retrieved_dp': ir_record_dp,
}
select_sent.select_end2end(**params)
def select_e2e():
params = {
'model_name': centrality_ensemble_config.CENTRALITY_MODEL_NAME_BASIC,
'n_iter': None,
'diversity_param_tuple':
centrality_ensemble_config.DIVERSITY_PARAM_TUPLE,
'cos_threshold': centrality_ensemble_config.
COS_THRESHOLD, # do not pos cosine similarity criterion?
'retrieved_dp': qa_record_dp,
}
select_sent.select_end2end(**params)
def select_end2end(model_name, n_iter=None, omega=10, save_out_fp=None):
params = {
'model_name': model_name,
'n_iter': n_iter,
'cos_threshold': 1.0, # do not pos cosine similarity criterion
'omega': omega,
}
output = select_sent.select_end2end(**params)
# output = rouge.compute_rouge_end2end(**params)
if save_out_fp:
content = '\t'.join((str(omega), output))
with io.open(save_out_fp, encoding='utf-8', mode='a') as f:
f.write(content + '\n')