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

Question

Apple 最近发布了一个允许创建机器学习模型的框架。我对表格数据感兴趣，但我没有在网上找到任何示例。谁能提供一段有效的代码？我尝试了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 行引发了以下错误：

错误：MyPlaygroundu.playground:9:22：错误：使用未解析的标识符“URL”尝试 pricer.write（到：URL（fileURLWithPath：“/Users/createml/HousePricer.mlmodel”））

Answer 1

URL来自Foundation框架。

添加：

import Foundation

就在之前import CreateML。