Core ML 3 现在允许我们进行设备上的训练。在创建一个可更新的Core ML模型后,我们需要使用MLUpdateTask
函数在设备上更新它,该函数需要三个参数:Model URL
、MLBatchProvider
和MLModelConfiguration
。
由于Core ML 3刚刚发布,其文档非常有限,特别是关于如何准备训练数据或MLBatchProvider
的说明。
问题:如何准备训练数据或创建MLBatchProvider
?
回答:
如果你的模型名为TestModel
,那么应该有一个名为TestModelTrainingInput
的类可用。
let singleTrainingData = try TestModelTrainingInput(input: .init[1,2,3], output_true: .init([4,5,6]))let trainingData = MLArrayBatchProvider(array: [singleTrainingData])