def generation_subject_predicate_object_triple_strategy_two(
standard_model_output):
result_json_write_f = open("subject_predicate_object_predict_output.json",
"w",
encoding='utf-8')
for text, entity_tuple_list, predicate_head in standard_model_output.text_entity_predicate_generator(
):
predicate_head_id_matrix = get_predicate_matrix(predicate_head)
print(text) #《不是所有时光都微笑》是2012年7月1日光明日报出版社出版的书籍，作者是蓝瞳
print(
entity_tuple_list
) #[('图书作品', 2, '不是所有时光都微笑'), ('出版社', 19, '光明日报出版社'), ('人物', 35, '蓝瞳')]
print(predicate_head_id_matrix.shape) #(128, 128, 50)
print("\n")
#Writing strategy function
# pass
# return spo_list = "spo_list": [{"predicate": "毕业院校", "object_type": "学校", "subject_type": "人物", "object": "四川外语学院", "subject": "王润泽"}]
line_dict = dict()
line_dict["text"] = text
line_dict["spo_list"] = None
line_json = json.dumps(line_dict, ensure_ascii=False)
result_json_write_f.write(line_json + "\n")
result_json_write_f.close()
def get_candicate_text_and_spo_list(model_output_tuple):
text, entity_tuple_list, predicate_head = model_output_tuple[
0], model_output_tuple[1], model_output_tuple[2]
candicate_spo_list = list()
predicate_head_id_matrix = get_predicate_matrix(predicate_head)
for subject_entity_type, subject_entity_position, subject_entity_value in entity_tuple_list:
for object_entity_type, object_entity_position, object_entity_value in entity_tuple_list:
if subject_entity_type in subject_type_list and object_entity_type in object_type_list:
if (subject_entity_type, object_entity_type
) in schemas_subject_object_type_2_predicate_list:
correct_predicate_scope_list = schemas_subject_object_type_2_predicate_list[
(subject_entity_type, object_entity_type)]
subject_object_predicate_row = predicate_head_id_matrix[
subject_entity_position, object_entity_position]
candicate_predicate_list = get_candicate_predicate_list(
correct_predicate_scope_list,
subject_object_predicate_row)
for candicate_predicate in candicate_predicate_list:
candicate_spo_list.append({
"subject":
subject_entity_value,
"predicate":
candicate_predicate,
"object":
object_entity_value,
"subject_type":
subject_entity_type,
"object_type":
object_entity_type
})
return (text, candicate_spo_list)