需要一个用Pandas数据帧计算RMSE的简单示例。如果函数在循环中返回真值和预测值:
def fun (data):
...
return trueVal, predVal
for data in set:
fun(data)
然后一些代码将这些结果放在以下数据框中,其中x
是实际值,p
是预测值:
In [20]: d
Out[20]: {'p': [1, 10, 4, 5, 5], 'x': [1, 2, 3, 4, 5]}
In [21]: df = pd.DataFrame(d)
In [22]: df
Out[22]:
p x
0 1 1
1 10 2
2 4 3
3 5 4
4 5 5
问题:
1)如何将fun
函数的结果放入df
数据帧中?
2)如何使用df
数据帧计算RMSE?
问题1
这取决于数据的格式。我希望你已经有了真正的价值,所以这个函数只是一个传递。
问题2
与
pandas
((df.p - df.x) ** 2).mean() ** .5
与
numpy
(np.diff(df.values) ** 2).mean() ** .5
相关问题 更多 >
编程相关推荐