将在Sklearn上训练的线性回归模型转换为ONNX模型

时间:2018-09-25 23:34:31

标签: scikit-learn onnx onnx-coreml

我打算将经过Sklearn训练的线性模型转换为Onnx模型。我使用以下代码:

从onnxmltools.convert.common.data_types导入FloatTensorType,Int64TensorType

onnx_model = onnxmltools.convert_sklearn(myTrainedModel, name=None, initial_types = [('float_input', FloatTensorType)], doc_string='linearRegression',targeted_onnx=onnx.__version__, custom_conversion_functions=None, custom_shape_calculators=None)

我有9个功能,但出现以下错误:

onnx_model = onnxmltools.convert_sklearn(myTrainedModel, name=None, 
initial_types = [('float_input', FloatTensorType)], doc_string='linearRegression',targeted_onnx=onnx.__version__, custom_conversion_functions=None, custom_shape_calculators=None)

对于这种类型的转换,没有明确的例子,我对此感到非常困惑……有什么帮助吗?

0 个答案:

没有答案