如何在Swift CoreML中使MLModel可更新？

import CreateML

// 假设你有一个CSV文件，其中包含训练数据
let data = try MLDataTable(contentsOf: URL(fileURLWithPath: "path/to/your/data.csv"))

// 创建一个分类器模型
let classifier = try MLTextClassifier(trainingData: data, textColumn: "text", labelColumn: "label")

// 保存模型，并使其可更新
let metadata = MLModelMetadata(author: "Your Name", shortDescription: "A text classifier model", version: "1.0")
try classifier.write(to: URL(fileURLWithPath: "path/to/save/YourModel.mlmodel"), metadata: metadata, options: .init(isUpdatable: true))

import CoreML

// 加载模型
guard let modelURL = Bundle.main.url(forResource: "YourModel", withExtension: "mlmodelc") else {
    fatalError("Model file not found")
}

let model = try! MLModel(contentsOf: modelURL)

// 创建一个MLUpdateTask来更新模型
let updateURL = Bundle.main.url(forResource: "updateData", withExtension: "json")!
let updateData = try! MLDataTable(contentsOf: updateURL)

let updateTask = try! MLUpdateTask(forModelAt: modelURL, trainingData: updateData, configuration: nil, completionHandler: { context in
    if context.task.state == .completed {
        print("Model updated successfully")
        // 保存更新后的模型
        let updatedModelURL = FileManager.default.temporaryDirectory.appendingPathComponent("UpdatedModel.mlmodelc")
        try! context.model.write(to: updatedModelURL)
    } else {
        print("Model update failed")
    }
})

// 开始更新任务
updateTask.resume()

// 加载更新后的模型
let updatedModelURL = FileManager.default.temporaryDirectory.appendingPathComponent("UpdatedModel.mlmodelc")
let updatedModel = try! MLModel(contentsOf: updatedModelURL)

// 使用更新后的模型进行预测
let input = YourModelInput(text: "Some input text")
let prediction = try! updatedModel.prediction(from: input)
print(prediction)

let configuration = MLModelConfiguration()
configuration.parameters = [
    .learningRate: 0.01
]

let updateTask = try! MLUpdateTask(forModelAt: modelURL, trainingData: updateData, configuration: configuration, completionHandler: { context in
    // 处理更新结果
})