PredictionBuilder
PredictionBuilder 是一个使用线性回归构建预测的机器学习库。
安装
CocoaPods
要使用 CocoaPods 安装 PredictionBuilder,请将以下行添加到您的 Podfile
pod 'PredictionBuilder', '~> 1.2.0'
Carthage
要使用 Carthage 安装 PredictionBuilder,请将以下行添加到您的 Cartfile
github "denissimon/prediction-builder-swift"
Swift 包管理器
要使用 Swift 包管理器 安装 PredictionBuilder,请将以下内容添加到您的 Package.swift
文件中:
dependencies: [
.Package(url: "https://github.com/denissimon/prediction-builder-swift.git", majorVersion: 1)
]
手动安装
将 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 许可下授权