自定义 MLModel 文件

Question

我是机器学习新手，想进行以下实现

• 想要创建自定义 .mlmodel，输入为“此文件的 xls 或 csv 或 nsdata”，输出应为双精度或数组。
• Pythone 文件应该能够读取输入数据，因为我将使用来自该输入的 train_data。
• Pythone 将对此输入数据进行一些计算并对此提供预测（我将使用 sklearn、LinearRegression 进行此计算）

谁能帮我解决这个问题？

Answer 1

您可以按照您的建议使用 python 通过 SKLearn 训练您的模型。这是一篇很好的入门文章（请确保您使用的是 Sklearn 而不是 Statsmodels）。

https://towardsdatascience.com/simple-and-multiple-linear-regression-in-python-c928425168f9

当您训练好模型后，您可以使用 Apple 的 coremltools 对其进行转换：

https://github.com/apple/coremltools

转换完成后，您可以将 .mlmodel 文件添加到您的 xcode 项目中。然后，您需要编写一些代码来获取从您的应用收集的所有模型输入，并将它们作为输入传递给模型。

祝你好运！