Update logic2.py

logic2.py

@@ -19,6 +19,13 @@ import yaml
 
 
 import pickle
+
+def net_repr_html(self):
+    nodes, edges, height, width, options = self.get_network_data()
+    html = self.template.render(height=height, width=width, nodes=nodes, edges=edges, options=options)
+    return html
+
+Network._repr_html_ = net_repr_html
 #----------------------------------------------
 # SAGE model
 class GNNEncoder(torch.nn.Module):
@@ -101,6 +108,7 @@ def get_recommendation(model,data,user_id):
     top_ten_recs = [get_title(movie_idx) for movie_idx in top_ten_recs]
     return {'user': user_id, 'rec_movies': top_ten_recs}
 
+
 def make_1_hop_graph(data,user_id):
     a = data["user", "rates", "movie"].edge_index
     b = data["user", "rates", "movie"].edge_label