feat: calculate average loss

Author: ChenZiHong-Gavin

judge.py

@@ -10,6 +10,20 @@
 
 load_dotenv()
 
+def calculate_average_loss(graph: NetworkXStorage):
+    """
+    Calculate the average loss of the graph.
+
+    :param graph: NetworkXStorage
+    :return: float
+    """
+    edges = asyncio.run(graph.get_all_edges())
+    total_loss = 0
+    for edge in edges:
+        total_loss += edge[2]['loss']
+    return total_loss / len(edges)
+
+
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
@@ -38,3 +52,6 @@
     graph_file = asyncio.run(graph_storage.get_graph())
 
     new_graph.write_nx_graph(graph_file, args.output)
+
+    average_loss = calculate_average_loss(graph_storage)
+    print(f"Average loss of the graph: {average_loss}")

scripts/judge.sh

@@ -1 +1 @@
-python3 evaluate.py --output cache/output/new_graph.graphml \
+python3 judge.py --output cache/output/new_graph.graphml \