Я намерен преобразовать линейную модель, которая обучается на Sklearn, в модель Onnx.Я использую следующий код:
из onnxmltools.convert.common.data_types import 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)
Нет четкого примера для этого типа преобразования, и я действительно смущен этим ... любая помощь?