Swift中使用MLRegressor

时间:2019-12-28 13:33:51

标签: swift dataset regression

我对这项挑战有些困惑,因此决定尝试在这里寻求帮助。

我正在使用Swift开发iOS应用。我用Daniel Gindi's Charts绘制图表(出于上下文,与我的问题无关)。

我有以下数据集:

let data = [(key: -15.64, value: -2.90996446628), (key: -9.66, value: -0.42763685848), (key: -2.43, value: -2.593804963), (key: -2.188, value: -3.21188185361), (key: -0.65, value: -0.64038615264), (key: -0.63, value: 0.54083188026), (key: 1.07, value: -0.73129770794), (key: 1.12, value: -0.64902953727), (key: 1.27, value: -0.57766476054), (key: 2.57, value: 0.4391731834), (key: 2.77, value: -0.82565483197), (key: 16.32, value: -0.78957840951)]

这是我从服务器接收到的数据输入中的一组键值对。目前,我使用它来创建散点图:

Scatter chart

现在,我想添加一条回归线。苹果公司在其web page上展示了一个提及MLRegressor的示例。具体来说,这张图片:

MLRegressor example

但是,我找不到可以帮助我建立自己的线性回归线的教程。我真正需要的是通过/通过MLRegressor为数据集中的每个键值对生成一个数据点数组。然后,我可以使用这个新的数据点数组构建折线图,并将其添加到散点图的顶部。

我们非常感谢您使用MLRegressor生成数据数组!

1 个答案:

答案 0 :(得分:0)

在第一次投票失败后感到沮丧之后,我不得不做更多的搜索,这就是我发现(并实现)的内容。 RayWenderlich's GitHub上有一套很棒的算法。

此外,Swift黑客技术的Paul Hudson提供了很好的基础intro to Create ML

在这两种资源之间,我实现了我所寻找的目标,并且所得到的预测与我的导师(华尔街人)手动创建的预测非常匹配。

除了显示混合数据图的方法外,该代码实际上很短。但是,CombinedChartView也有很好的资源。