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

Question

苹果最近发布了一个框架，允许创建机器学习模型。我对表格数据感兴趣，但我还没有在网上找到任何例子。有谁能提供一段有用的代码吗？我尝试了苹果公司提供的以下内容，但没有成功：

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: error:使用未解析的标识符'URL‘尝试pricer.write(to: URL(fileURLWithPath："/Users/createml/HousePricer.mlmodel"))

Answer 1

URL来自Foundation框架。

添加：

import Foundation

就在import CreateML之前。