Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Abodi-Massarwa committed Aug 16, 2024
1 parent 643bb9d commit 6c7b549
Showing 1 changed file with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1250,9 +1250,9 @@ def helper_generate_directed_graph_base64(graph, seed=42,category:str='',iterati
#logger.info(f'**********\nRunning helper_generate_directed_graph_base64\n**********')

plt.figure()
plt.title('Envy Graph')
additional_text=f'{category} iteration {iteration}'
plt.figtext(0.5, 0.95, additional_text, wrap=True, horizontalalignment='center', fontsize=10)
plt.title('Envy Graph',fontsize=16)
additional_text=f'category -> {category} iteration -> {iteration}'
plt.figtext(0.5, 0.90, additional_text, wrap=True, horizontalalignment='center', fontsize=10)
pos = nx.spring_layout(graph, seed=seed) # Use a seed for reproducibility
nx.draw(graph, pos, with_labels=True, node_color='lightblue', edge_color='gray', node_size=500, font_size=10, arrows=True)

Expand All @@ -1270,8 +1270,8 @@ def helper_generate_bipartite_graph_base64(graph,iteration:int,category:str):
#logger.info(f'**********\nRunning helper_generate_bipartite_graph_base64\n**********')
plt.figure()
plt.title('Agent-Item Bipartite Graph', fontsize=16)
additional_text=f'{category} iteration {iteration}'
plt.figtext(0.5, 0.95, additional_text, wrap=True, horizontalalignment='center', fontsize=10)
additional_text=f'category -> {category} iteration -> {iteration}'
plt.figtext(0.5, 0.90, additional_text, wrap=True, horizontalalignment='center', fontsize=10)
try:
top_nodes = {n for n, d in graph.nodes(data=True) if d['bipartite'] == 0}
bottom_nodes = set(graph) - top_nodes
Expand Down

0 comments on commit 6c7b549

Please sign in to comment.