Spaces:
Sleeping
Sleeping
eaglelandsonce
commited on
Commit
•
db27fce
1
Parent(s):
0e91aec
Rename pages/17_Graph2.py to pages/17_Graphs2.py
Browse files
pages/{17_Graph2.py → 17_Graphs2.py}
RENAMED
@@ -67,14 +67,14 @@ def create_sample_graph():
|
|
67 |
}
|
68 |
)
|
69 |
|
70 |
-
return graph, graph_tensor
|
71 |
|
72 |
# Streamlit app
|
73 |
def main():
|
74 |
st.title("Graph Neural Network Architecture Visualization")
|
75 |
|
76 |
# Create sample graph
|
77 |
-
nx_graph, graph_tensor = create_sample_graph()
|
78 |
|
79 |
# Create and compile the model
|
80 |
model = model_fn(graph_tensor.spec)
|
|
|
67 |
}
|
68 |
)
|
69 |
|
70 |
+
return graph, graph_tensor, node_features, edge_features
|
71 |
|
72 |
# Streamlit app
|
73 |
def main():
|
74 |
st.title("Graph Neural Network Architecture Visualization")
|
75 |
|
76 |
# Create sample graph
|
77 |
+
nx_graph, graph_tensor, node_features, edge_features = create_sample_graph()
|
78 |
|
79 |
# Create and compile the model
|
80 |
model = model_fn(graph_tensor.spec)
|