arushij1711

def convert_to_onnx(model, data): initial_type = [('float_input', FloatTensorType([None, **data.shape[0]**]))] onnx_model = convert_sklearn(model, initial_types=initial_type) print("onnx_model.type:",type(onnx_model)) mlflow.set_tag("onnx_version",onnx.__version__) return onnx_model It should be data.shape[1]