自定义MLModel文件

Question

我是机器学习的新手，想要做以下的实施

• 想要创建一个自定义.mlmodel，输入“xls或csv或此文件的nsdata”，输出应为double或array。
• 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项目中。然后，您需要编写一些代码来获取从您的应用收集的所有模型输入，并将它们作为输入传递给模型。

祝你好运！