Krish Patel commited on
Commit
25bb7ca
·
1 Parent(s): 60631d5

Enhanced knowledge graph

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def generate_knowledge_graph_viz(text, nlp, tokenizer, model):
44
 
45
  # Randomly select subset of edges (e.g. 60% of edges)
46
  edges = list(kg_builder.knowledge_graph.edges())
47
- selected_edges = random.sample(edges, k=int(len(edges) * 0.3))
48
 
49
  # Create a new graph with selected edges
50
  selected_graph = nx.DiGraph()
 
44
 
45
  # Randomly select subset of edges (e.g. 60% of edges)
46
  edges = list(kg_builder.knowledge_graph.edges())
47
+ selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
48
 
49
  # Create a new graph with selected edges
50
  selected_graph = nx.DiGraph()