add for both prfs

terrierteam · Dec 6, 2024 · 8459f05 · 8459f05
1 parent 6f9ae27
commit 8459f05
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyterrier_dr/prf.py b/pyterrier_dr/prf.py
@@ -79,6 +79,9 @@ def __init__(self,
     ):
         self.k = k
 
+    def compile(self) -> pt.Transformer:
+        return pt.RankCutoff(self.k) >> self
+
     @pta.transform.by_query(add_ranks=False)
     def transform(self, inp: pd.DataFrame) -> pd.DataFrame:
         """Performs Average PRF on the input dataframe."""