Core ML, как программно создать обновляемую модель? - PullRequest
Новая
       16

Core ML, как программно создать обновляемую модель?

0 голосов
/ 11 июля 2019

Я программно создал модель, используя игровую площадку на Mac, но модель не имеет параметров обновления. Кто-нибудь смог создать обновляемую модель? Я не вижу каких-либо изменений API для этого.

Созданы модели с использованием Catalina и Xcode 11 beta 3, но когда модель загружается в проект, сгенерированный класс не показывает модель как обновляемую. 

        if let dataTable = try? MLDataTable(dictionary: priorityData) {
            let (trainingCSVData, testCSVData) = dataTable.randomSplit(by: 0.8, seed: 0)
            print( "Training Data = " + trainingCSVData.description)

            do {
                let deleteProbabilityClassifier = try MLClassifier(trainingData: dataTable, targetColumn: "DeleteProbability")
                // evaluate it
                let metrics = deleteProbabilityClassifier.evaluation(on: testCSVData)

                let modelMetadata = MLModelMetadata(author: "Me", shortDescription: "Model for DeleteProbability Prediction", version: "1.1")
                let modelPath = homeDirURL.appendingPathComponent(name + "-deleteProbabilityPredictor.mlmodel", isDirectory: false)
                do {
                    try deleteProbabilityClassifier.write(to: modelPath, metadata: modelMetadata)
                    print("Success - DeleteProbability Model written" + modelPath.absoluteString )
                    print(metrics)
                    return true
                }
                catch {
                    print("Failure to write DeleteProbabilityClassifier - " + modelPath.absoluteString)
                }
            }
            catch {
                print("ML Classifier failed")
                print("ML Classsifier threw error = " + error.localizedDescription)
            }
        }
        else {
            print("Failed - MLDataTable ")
            print(type(of: priorityData))
        }

Надеемся построить обновляемую модель.

...