使用 CreateML 在 Swift 中为机器学习创建数据表

Question

Apple 最近发布了一个允许创建机器学习模型的框架。我对 tabular data 感兴趣，但我没有在网上找到任何示例。谁能提供一段有效的代码？我尝试了 Apple 提供的以下方法但没有成功：

import CreateML

// Specify Data
let trainingCSV = URL(fileURLWithPath: "/Users/createml/HouseData.csv")
let houseData = MLDataTable(contentsOf: trainingCSV)
let (trainingData,testData) = houseData.randomSplit(by: 0.8, seed: 0)

// Create Model
let pricer = try MLRegressor(trainingData: houseData, targetColumn: "price")

// Evaluate Model
let metrics = try pricer.testingMetrics(on: testData)

// Save Model
try pricer.write(to: URL(fileURLWithPath: "/Users/createml/HousePricer.mlmodel"))

特别是这段代码在第 4 行和第 15 行抛出以下错误：

error: MyPlaygroundu.playground:9:22: error: use of unresolved identifier 'URL' try pricer.write(to: URL(fileURLWithPath: "/Users/createml/HousePricer.mlmodel"))

Answer 1

URL 来自 Foundation 框架。

添加：

import Foundation

就在 import CreateML.

之前

使用 CreateML 在 Swift 中为机器学习创建数据表

Creating Data Tables for Machine Learning in Swift using CreateML

machine-learning

swift

createml