Add PageRank

app.py

@@ -15,9 +15,10 @@
 __author__ = 'Dmitry Ustalov'
 __license__ = 'Apache 2.0'
 
-from typing import IO, Tuple, List, cast
+from typing import IO, Tuple, List, cast, Dict
 
 import gradio as gr
+import networkx as nx
 import numpy as np
 import numpy.typing as npt
 import pandas as pd
@@ -67,6 +68,19 @@ def bradley_terry(wins: npt.NDArray[np.int64], ties: npt.NDArray[np.int64],
     return p
 
 
+def pagerank(wins: npt.NDArray[np.int64], ties: npt.NDArray[np.int64],
+             seed: int = 0, tolerance: float = 10e-6, limit: int = 100) -> npt.NDArray[np.float64]:
+    A = wins + .5 * ties
+
+    G = nx.from_numpy_array(A, create_using=nx.DiGraph)
+
+    pagerank: Dict[int, float] = nx.algorithms.pagerank(G, max_iter=limit, tol=tolerance)
+
+    scores = np.array([pagerank[i] for i in range(len(G))])
+
+    return scores
+
+
 # https://gist.github.com/dustalov/41678b70c40ba5a55430fa5e77b121d9#file-newman-py
 def newman(wins: npt.NDArray[np.int64], ties: npt.NDArray[np.int64],
            seed: int = 0, tolerance: float = 10e-6, limit: int = 20) -> npt.NDArray[np.float64]:
@@ -117,6 +131,7 @@ def newman(wins: npt.NDArray[np.int64], ties: npt.NDArray[np.int64],
 
 ALGORITHMS = {
     'Bradley-Terry (1952)': bradley_terry,
+    'PageRank': pagerank,
     'Newman (2023)': newman,
 }
 

requirements.txt

@@ -1 +1,2 @@
+networkx
 plotly_express