Создание таблиц данных для машинного обучения в Swift с использованием CreateML - PullRequest
0 голосов
/ 04 октября 2018

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:

error: MyPlaygroundu.playground:9:22: ошибка: использование неразрешенного идентификатора 'URL', попробуйте pricer.write (to: URL (fileURLWithPath: "/Users/createml/HousePricer.mlmodel"))

1 Ответ

0 голосов
/ 04 октября 2018

URL из структуры Foundation.

Добавить:

import Foundation

непосредственно перед import CreateML.

...