我打算将经过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)
对于这种类型的转换,没有明确的例子,我对此感到非常困惑……有什么帮助吗?