Skip to content
Snippets Groups Projects
Commit ee1ca829 authored by vvye's avatar vvye
Browse files

Fix weighted pagerank

parent df1cc4c8
No related branches found
No related tags found
No related merge requests found
from collections import Counter
from datetime import datetime
import random
import igraph
......@@ -34,13 +35,13 @@ def rank_dates_by_wilson(articles, start_date, end_date):
edges.append((pub_date, mentioned_date, edge_weight))
# create a graph from the edge list
g = igraph.Graph.TupleList(edges, directed=True)
g = igraph.Graph.TupleList(edges, directed=True, edge_attrs='weight')
vertex_names = g.vs['name']
# igraph.plot(g, layout='kk', vertex_label=g.vs['name'], bbox=(3000, 3000))
# rank vertices by pagerank score
pagerank_scores = g.pagerank()
pagerank_scores = g.pagerank(directed=True, weights=g.es['weight'])
ranked_dates = util.rank(vertex_names, scores=pagerank_scores)
return ranked_dates
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment