[Tools] MlResponse: assure equality of number of checked features to that in the .onnx file (AliceO2Group#16424)

Co-authored-by: Vít Kučera <26327373+vkucera@users.noreply.github.com>

Author: 2 people

Tools/ML/MlResponse.h

@@ -151,8 +151,14 @@ class MlResponse
   void init(bool enableOptimizations = false, int threads = 0)
   {
     uint8_t counterModel{0};
+    const int numCachedIndices = static_cast<int>(mCachedIndices.size());
     for (const auto& path : mPaths) {
       mModels[counterModel].initModel(path, enableOptimizations, threads);
+      const int numInputNodes = mModels[counterModel].getNumInputNodes();
+      if (numInputNodes != numCachedIndices) {
+        LOG(fatal) << "Number of input nodes in the model " << path << " is different from the number of input features indices (" << numInputNodes << " vs " << numCachedIndices << ")";
+        return;
+      }
       ++counterModel;
     }
   }