计算均方根值和graphlab中的平均值

时间:2016-05-11 09:28:03

标签: graphlab

我有一个格式为cp = [cars [' car_models'] ==" chevy"]的数据框 如何获得cp的平均值?

到处寻找如何做到这一点。谢谢你的帮助

1 个答案:

答案 0 :(得分:0)

正如@papayawarrior所说,SFrame没有平均值。 SArrays(float / int类型)可能有这些。

>>> sf = gl.SFrame({"x":[1,2,3]}) # SFrame with a single column (SArray) x.
>>> sf["x"].mean() # sf["x"] grabs the SArray x, then we take its average.
2.0

如果你想要均方根错误,你还应该有两个SArrays(可能在同一个SFrame中)。我不知道" root均值的平方值"是

import graphlab as gl

cars = gl.SFrame({
    "car_models": ["chevy", "ford", "chevy"],
    "targets": [1, 2, 3],
    "predictions": [7, 9, 8]
    })

cp = cars[cars["car_models"] == "chevy"]
rmse = gl.evaluation.rmse(cp["targets"], cp["predictions"])

此示例中的rmse为5.522680508593631。