RSS 2019-12 - 1/1 Blog benchmark (4) onnx (8)


2019-12 - 1/1#

Custom C++ TopK#

2019-12-16

It started with the fact the python runtime for the AdaBoostRegressor was quite slow. I noticed three operators were quite slow even though their implementation was based on numpy: TopK, ArrayFeatureExtractor and GatherElement. I made a custom implementation of the first two.

post

RandomForestClassifier - prediction for one observation#

2019-12-04

I was meeting with Olivier Grisel this morning and we were wondering why scikit-learn was slow to compute the prediction of a random forest for one observation compare to what onnxruntime does, and more specically some optimized C++ code inspired from onnxruntime. We used py-spy and wrote the following script:

post


RSS 2019-12 - 1/1 2021-08 (3) 2022-02 (1) 2022-05 (1) 2022-06 (1) 2022-11 (1)