Methods

Summary

method

class parent

truncated documentation

__init__

BenchPerf

__init__

BenchPerfTest

__init__

ProfilerCall

__init__

OnnxRuntimeBenchPerfTest

__iter__

ProfilerCall

Iterates on stored profiled. Returns a couple (profile, configuration).

__len__

ProfilerCall

Returns the number of stored profiles.

__repr__

BenchPerf

usual

_get_random_dataset

OnnxRuntimeBenchPerfTest

Returns a random datasets.

_get_random_dataset

OnnxRuntimeBenchPerfTestBinaryClassification

Returns a random datasets.

_get_random_dataset

OnnxRuntimeBenchPerfTestRegression

Returns a random datasets.

_init

OnnxRuntimeBenchPerfTest

Finalizes the init.

data

BenchPerfTest

Generates one testing dataset.

data

OnnxRuntimeBenchPerfTest

Generates random features.

dump_error

BenchPerfTest

Dumps everything which is needed to investigate an error. Everything is pickled in the current folder or dump_folder

enumerate_run_benchs

BenchPerf

Runs the benchmark.

enumerate_tests

BenchPerf

Enumerates all possible options.

extract_model_info_onnx

OnnxRuntimeBenchPerfTest

Populates member self.onnx_info with additional information on the ONNX graph.

extract_model_info_skl

OnnxRuntimeBenchPerfTest

Populates member self.skl_info with additional information on the model such as the number of node for …

fct_filter_test

BenchPerf

Tells if the test by conf is valid or not.

fcts

BenchPerfTest

Returns the function call to test, it produces a dictionary {name: fct} where name is the name of …

fcts

OnnxRuntimeBenchPerfTestBinaryClassification

Returns a few functions, tests methods perdict, predict_proba for both scikit-learn and OnnxInference

fcts

OnnxRuntimeBenchPerfTestRegression

Returns a few functions, tests methods perdict, predict_proba for both scikit-learn and OnnxInference

match

ProfilerCall

Tells if the profiler should be run on this set of parameters.

model_info

OnnxRuntimeBenchPerfTest

Returns additional informations about a model.

profile

BenchPerf

Checks if a profiler applies on this set of parameters, then profiles function fct.

profile

ProfilerCall

Profiles function fct, calls it repeat times.

to_txt

ProfilerCall

Saves all profiles into one file.

validate

BenchPerfTest

Runs validations after the test was done to make sure it was valid.

validate

OnnxRuntimeBenchPerfTest

Checks that methods predict and predict_proba returns the same results for both scikit-learn and …