PredictionBuilder

PredictionBuilder 是一个用于机器学习的库，它使用线性回归来构建预测模型。

要使用 Swift Package Manager 安装 PredictionBuilder，请按照以下步骤操作。

Xcode: File -> Add Packages
Enter Package URL: https://github.com/denissimon/prediction-builder-swift

要使用 CocoaPods 安装 PredictionBuilder，请将以下行添加到您的 Podfile 中。

pod 'PredictionBuilder', '~> 1.2'

要使用 Carthage 安装 PredictionBuilder，请将以下行添加到您的 Cartfile 中。

github "denissimon/prediction-builder-swift"

将 PredictionBuilder.swift 文件复制到您的项目中。

import PredictionBuilder

let data: [[Double]] = [[1,20],[2,70],[2,45],[3,81],[5,73],[6,80],[7,110]]
let x: Double = 4.5

let prediction = PredictionBuilder()

prediction.set(x: x, data: data)

// What is the expected y value for a given x value?
do {
    let result = try prediction.build() // y = 76.65
} catch let msg {
    print(msg)
}

返回的实例具有以下属性

result.lnModel 拟合数据的线性模型: "29.56362+10.46364x"

result.cor 相关系数: 0.8348

result.x 给定的 x 值: 4.5

result.y 预测的 y 值: 76.65

根据 MIT 许可证 授权