如何解决如何将 MLPC 从 SKLearn 转换为 CoreML
我目前正在开展一个项目,为一门课程训练、构建和部署机器学习模型。我已经训练了以下模型:
# Modelo Nº 5
from sklearn.neural_network import MLPClassifier
model_5 = MLPClassifier(hidden_layer_sizes = (TOTAL_INPUTS,),activation = 'relu',solver = 'adam',batch_size = 1000,verbose = True,early_stopping = True,n_iter_no_change = 5,max_iter = 100,validation_fraction = 0.2)
model_5.fit(x_train,y_train.reshape((y_train.shape[0],)))
我想知道是否可以将这个训练好的模型转换为 CoreML 的 .mlmodel 格式,然后在一个简单的 iOS 应用程序中使用它。
我尝试了以下代码:
import coremltools
coreml_model = coremltools.converters.sklearn.convert(model_5)
coreml_model.short_description = "CIFAR10 MLPC SKLearn Classifier"
coreml_model.save("cifar10_sklearn_mlpc.mlmodel")
但我收到此错误:
提前致谢
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。