首页 > 解决方案 > 为什么使用 turicreate 和 export_coreml 到 .mlmodel 文件它是一个文件夹

问题描述

我正在使用 turicreate 来训练核心 ml 模型(苹果)我的代码是:

import os
import turicreate as tc

data = tc.image_analysis.load_images('Train', with_path=True)
data['label'] = data['path'].apply(lambda path: os.path.basename(os.path.dirname(path)))
data.save('my-model.sframe')

data =  tc.SFrame('my-model.sframe')

# Make a train-test split
train_data, test_data = data.random_split(0.8)

# Create the model
model = tc.image_classifier.create(train_data, target='label')

# Save predictions to an SArray
predictions = model.predict(test_data)

# Evaluate the model and print the results
metrics = model.evaluate(test_data)
print(metrics['accuracy'])

# Save the model for later use in Turi Create
model.save('my-model.model')

# Export for use in Core ML
model.export_coreml('my-model.mlmodel')

和结果:我有 3 个文件夹:my-model.sframe、my-model.model、my-model.mlmodel

我尝试压缩文件夹 my-model.mlmodel 并重命名为文件:my-model.mlmodel,但 xcode 无法读取它。

我如何修复它?我想要文件 my-model.mlmodel,xcode 可以读取它,而不是文件夹

我的数据列是:7k 标签和 1,3M 文件,我无法使用 Create ML 工具

谢谢你

标签: pythoncoremlcoremltoolsturi-create

解决方案


推荐阅读