具有DataFrames的RankMetrics（）PySpark

Question

+-----+------+---------+----------+
|Rank  USER  |ITEM     |prediction|
+-----+------+---------+----------+
|13   |548.0 |12.0     |2.3577058 |
|6    |573.0 |12.0     |1.5444818 |
+-----+------+---------+----------+

我有一个来自ALS的预测数据帧，想知道如何将其转换为RankMetrics（）类接受的格式？我知道我需要一个RDD，它是一个列表列表？但似乎需要特殊格式吗？我尝试使用来自How to convert a DataFrame back to normal RDD in pyspark?的建议。