| import numpy as np |
| import matplotlib.pyplot as plt |
| import numpy as np |
|
|
| def find_parent(matrix, node): |
| parents = matrix[:, node] |
| max_parent = np.argmax(parents) |
| if parents[max_parent] > 0: |
| return max_parent |
| else: |
| return None |
|
|
|
|
| def find_tree(matrix, node, depth=1, children=[], max_depth=1, visited=set()): |
| result = [] |
| parent = find_parent(matrix, node) |
| if parent is not None and parent not in visited: |
| result.append([parent, node]) |
| for child in children: |
| result.append([node, child]) |
| if depth < max_depth: |
| visited.add(node) |
| result.extend(find_tree(matrix, parent, depth + 1, visited=visited)) |
| |
| return result |
|
|
| |
| def find_prob(tree, matrix): |
| prob = 1 |
| for parent, child in tree: |
| prob *= matrix[parent][child] |
| return prob |
|
|
| def find_forests(matrix, k): |
| forests = {} |
| for i in range(len(matrix)): |
| children = matrix[i] |
| child_list = [] |
| for j in range(len(children)): |
| if children[j] > 0: |
| child_list.append(j) |
| tree = find_tree(matrix, i, children=child_list) |
| tree = tuple([tuple(x) for x in tree]) |
| if tree: |
| prob = find_prob(tree, matrix) |
| if tuple(tree) in forests: |
| forests[tuple(tree)] += prob |
| else: |
| forests[tuple(tree)] = prob |
| sorted_forests = sorted(forests.items(), key=lambda x: x[1], reverse=True) |
| forest, prob = sorted_forests[0] |
| result = {} |
| |
| for parent, child in forest: |
| |
| if parent in result: |
| result[parent].append(child) |
| else: |
| result[parent] = [child] |
| return result, prob |
| def passage_outline(matrix,sentences): |
|
|
| result, prob = find_forests(matrix, 1) |
| print(result, prob) |
| structure = {} |
| for each in result.keys(): |
| structure[each] =[sentences[i] for i in result[each]] |
| outl = [] |
| outline_list = [] |
| for key in sorted(structure.keys()): |
| outline_list.append(f"主题：") |
| outl.append(f"主题：\n") |
| for sentence in structure[key]: |
| outline_list.append(sentence) |
| outl.append(f"- {sentence}\n") |
| return outl,outline_list |
| if __name__ == "__main__": |
| matrix = np.array([[0.0 ,0.02124888, 0.10647043 ,0.09494194 ,0.0689209 ], |
| [0.01600688 ,0.0 ,0.05879448 ,0.0331325 , 0.0155093 ], |
| [0.01491911 ,0.01652437, 0.0, 0.04714563, 0.04577385], |
| [0.01699071 ,0.0313585 , 0.040299 ,0.0 ,0.014933 ], |
| [0.02308992 ,0.02791895 ,0.06547201, 0.08517842 ,0.0]]) |
| sentences = ["主题句子1", "主题句子2", "主题句子3", "主题句子4", "主题句子5"] |
| print(passage_outline(matrix,sentences)[0]) |
| |
