Update app.py

app.py

@@ -104,15 +104,15 @@ class Visualizer:
  if d <= thresh and (neighbor == entity2 or neighbor == entity1) and len(neighbors) < num_neighbors + 1:
  neighbors.add(entity1)
  neighbors.add(entity2)
- spring_tension = (thresh - d) * 10 # for vis purposes
+ spring_tension = (thresh - d) * 1000 # for vis purposes
  G.add_edge(entity1, entity2, weight=spring_tension)
  neighbors.remove(neighbor)
  for entity1, entity2, d in tqdm(normalized_distances):
  if entity2 in neighbors and entity1 in neighbors:
- spring_tension = thresh - d
+ spring_tension = (thresh - d) * 10
  G.add_edge(entity1, entity2, weight=spring_tension)
 
- pos = nx.spring_layout(G, weight="weight") # Positions for all nodes
+ pos = nx.spring_layout(G, weight="weight", iterations=200, threshold=1e-6) # Positions for all nodes
  edges = G.edges(data=True)
  nx.draw_networkx_nodes(G, pos, node_size=1, alpha=0.01)
  edges_connected_to_specific_node = [(u, v) for u, v in G.edges() if u == neighbor or v == neighbor]
@@ -121,7 +121,7 @@ class Visualizer:
  # nx.draw_networkx_edges(G, pos, edgelist=edges_not_connected_to_specific_node, edge_color='gray', alpha=0.1, width=1)
  for u, v, d in edges:
  if u == neighbor or v == neighbor:
- nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 10)), 2)}, font_color="red", alpha=0.4) # reverse modifications
+ nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 1000)) * 100, 2)}, font_color="red", alpha=0.4) # reverse modifications
  nx.draw_networkx_labels(G, pos, font_size=14, font_family='sans-serif', font_color='green')
  nx.draw_networkx_labels(G, pos, labels={neighbor: neighbor}, font_size=14, font_family='sans-serif', font_color='red')
  plt.title(f'Graph of {distance_type}')