fixed problems in recommendationengine class, updated README

This commit is contained in:
Tristan Lee
2022-08-09 09:45:40 -05:00
parent b18e5591fa
commit cb30169ece
6 changed files with 60 additions and 43 deletions

View File

@@ -24,7 +24,7 @@ if __name__ == '__main__':
engine = polyphemus.base.RecommendationEngine(channel_list= [CHANNEL_NAME])
weighted_edge_list, claim_id_to_video = engine.generate(iterations = 1)
weighted_edge_list, channels, claim_id_to_video = engine.generate(iterations = ITERATIONS)
G = nx.DiGraph()
G.add_weighted_edges_from(weighted_edge_list)