GodsDevProject commited on
Commit
2e97670
·
verified ·
1 Parent(s): e57e1ea

Delete entity_graph.py

Browse files
Files changed (1) hide show
  1. entity_graph.py +0 -19
entity_graph.py DELETED
@@ -1,19 +0,0 @@
1
- import networkx as nx
2
- from typing import List, Dict
3
-
4
- def build_entity_graph(docs: List[Dict]) -> Dict:
5
- G = nx.Graph()
6
-
7
- for d in docs:
8
- agency = d.get("agency", "Unknown")
9
- G.add_node(agency, group="agency")
10
-
11
- for token in d.get("content", "").split():
12
- if token.isupper() and len(token) > 2:
13
- G.add_node(token, group="entity")
14
- G.add_edge(agency, token)
15
-
16
- return {
17
- "nodes": [{"id": n, "group": G.nodes[n]["group"]} for n in G.nodes],
18
- "links": [{"source": u, "target": v} for u, v in G.edges]
19
- }