从PySpark中的两个不同数据帧中减去列的值以找到RMSE

我无法弄清楚。我正在尝试计算测试数据和预测数据之间的RMSE。

测试

col1    col2
 a        2 
 b        3

预测

col1   col2
 a       4 
 b       5

我正在尝试执行此测试（col2）-预测（col2）。那是

2-4 =-2
3-5 =-2

我试过了

test.select("col2").subtract(prediction.select("col2"))

但是我没有得到所需的结果。我正在尝试获取此结果以找到RMSE。spark中是否有内置功能来找到RMSE？

谢谢。