Как создать MLImageClassifier в Xcode 10

Question

Я пытаюсь построить машинное обучение - Image Recognition с использованием Create ML на Xcode 10.1 Playground, но у меня возникают некоторые проблемы с размещением моих данных в модели.

У меня естьпапка с изображениями, пронумерованными от 1 до 1336, и файл .csv с 2 столбцами (имя изображения и классификация изображения).

Я не знаю точно, как поместить это в модель.

У меня есть это до сих пор:

import Cocoa
import CreateML

let data = try MLDataTable(contentsOf: URL(fileURLWithPath: "/Users/x/Desktop/CoreML/project/file.csv"))

let(trainingData, testingData) = data.randomSplit(by: 0.8, seed: 1)

let Classifier = try MLImageClassifier *need help here*

let evaluationMetrics = sentimentClassifier.evaluation(on: testingData)

let evaluationAccuracy = (1 - evaluationMetrics.classificationError) * 100

let metaData = MLModelMetadata(author: "x", shortDescription: "Model", version: "1.0")

try classifier.write(to: URL(fileURLWithPath: "/Users/x/Desktop/CoreML/project/XClassifier.mlmodel"))

Answer 1

Я считаю, что невозможно наклеить ярлыки на MLImageClassifier через .csv или любой другой отдельный файл.У вас есть только два варианта: использовать имена файлов в качестве меток или использовать каталоги в качестве меток (вероятно, предпочтительнее в случае многих изображений):

let model = try MLImageClassifier(trainingData: .labeledDirectories(at: trainingDir))
let evaluation = model.evaluation(on: .labeledDirectories(at: testingDir))

Вам нужно будет поместить изображения в подкаталоги, названные как метки в вашем .csv файл.