Skip to content

MVA improvements: multiple input nodes for tensorflow, ONNX runtime

Pieter David requested to merge piedavid/bamboo:mvaimprovements202011 into master

Partially addresses https://gitlab.cern.ch/cp3-cms/bamboo/-/issues/77

This addresses most of the points there (especially: multiple input nodes with different types, and thread safety with tensorflow).

What could still be added (deferred to a later PR)

  • XGBoost (for lack of a test file that can be read with the version of XGBoost in the LCG releases)
  • multiple inputs for PyTorch: should be possible, but I need to have a closer look at the documentation for how to get the shapes from the model then (I don't think anyone is using this yet, so this could be merged if the other changes are validated and more urgently needed)
Edited by Pieter David

Merge request reports