module testing.test_utils#

Short summary#

module mlprodict.testing.test_utils

Inspired from sklearn-onnx, handles two backends.

source on GitHub

Functions#

function

truncated documentation

create_tensor

Creates a tensor.

ort_version_greater

Tells if onnxruntime version is greater than ver.

Documentation#

Inspired from sklearn-onnx, handles two backends.

source on GitHub

mlprodict.testing.test_utils.create_tensor(N, C, H=None, W=None)#

Creates a tensor.

mlprodict.testing.test_utils.ort_version_greater(ver)#

Tells if onnxruntime version is greater than ver.

Parameters:

ver – version as a string

Returns:

boolean

source on GitHub