.

cw2.py

@@ -124,22 +124,26 @@ def also_like(data, doc_uuid, visitor_uuid, ascending, req_5=True):
 # Req 6
 def also_like_graph(data, doc_uuid, visitor_uuid, ascending):
     top_docs = also_like(data, doc_uuid, visitor_uuid, ascending, req_5=False)
-    y = top_docs[1].value_counts()
-    y = pd.DataFrame(y)
+    top_docs[0] = top_docs[0].str[-4:]
+    top_docs[1] = top_docs[1].str[-4:]
+    print(top_docs)
+    # y = top_docs[1].value_counts()
+    # y = pd.DataFrame(y)
     # y.loc[len(y.index)] = [visitor_uuid, doc_uuid]
-    print(y)
+    # print(y)
     dot = graphviz.Digraph()
     for index, row in top_docs.iterrows():
-        # if(index <= 7):
-        # print(row[0], row[1])
-        dot.node(str(row[0]), str(row[0])[-4:])
-        dot.node(str(row[1]), str(row[1])[-4:])
-        dot.edge(str(row[0]), str(row[1]))
-
-    # print(dot.source)
-
+        dot.node(str(row[0]), str(row[1]))
+    #     # if(index <= 7):
+    #     # print(row[0], row[1])
+    #     dot.node(str(row[0]), str(row[0])[-4:])
+    #     dot.node(str(row[1]), str(row[1])[-4:])
+    #     dot.edge(str(row[0]), str(row[1]))
+    #
+    # # print(dot.source)
+    #
     dot.render('output.dot').replace('\\', '/')
-
+    #
     (graph,) = pydot.graph_from_dot_file('output.dot')
     graph.write_png('output.png')
     return 'output.png'