CondFormats/MLObjects/interface/MLMetadataWrapper.h - master - githubmirror/cmssw

CondFormats/MLObjects/interface/MLMetadataWrapper.h
61 строка2 KB
hyejin kwon
ML Objects initialization
06 мар 2026, 18:01
06 мар 2026, 18:01343371d
#ifndef CondFormats_MLObjects_MLMetadataWrapper_h
#define CondFormats_MLObjects_MLMetadataWrapper_h

// -*- C++ -*-
//
// Package:     CondFormats/MLObjects
// Class  :     MLMetadataWrapper
//
/**
  \class MLMetadataWrapper MLMetadataWrapper.h "CondFormats/MLObjects/interface/MLMetadataWrapper.h"

  Description: persistent wrapper for ML model metadata and ONNX Runtime integration.
               Extends basic Condition Database fields (name, version, hash) with 
               extended metadata parsed from external JSON sources.

  Author:      H. Kwon
  Created:     Fri, 27 Feb 2026 12:34:38 CET
*/

#include <string>
#include <vector>
#include <memory>
#include "CondFormats/Serialization/interface/Serializable.h"
#include "PhysicsTools/ONNXRuntime/interface/ONNXRuntime.h"

class MLMetadataWrapper {
public:
  MLMetadataWrapper() = default;
  MLMetadataWrapper(
      std::string model_name, std::string version, std::string hash, std::string model_path, std::string preproc_path)
      : model_name_(model_name), version_(version), hash_(hash), model_path_(model_path), preproc_path_(preproc_path) {}

  std::string model_name() const { return model_name_; }
  std::string version() const { return version_; }
  std::string hash() const { return hash_; }
  std::string model_path() const { return model_path_; }
  std::string preprocessing_path() const { return preproc_path_; }
  const std::vector<std::string>& input_features() const { return input_features_; }
  const std::vector<std::string>& output_names() const { return output_names_; }

  void setOnnxRuntime(std::shared_ptr<cms::Ort::ONNXRuntime> runtime) { runtime_ = runtime; }

  std::shared_ptr<cms::Ort::ONNXRuntime> onnxRuntime() const { return runtime_; }

  void set_input_features(const std::vector<std::string>& features) { input_features_ = features; }
  void set_output_names(const std::vector<std::string>& names) { output_names_ = names; }

private:
  std::string model_name_;
  std::string version_;
  std::string hash_;
  std::string model_path_;
  std::string preproc_path_;
  std::vector<std::string> input_features_;
  std::vector<std::string> output_names_;
  std::shared_ptr<cms::Ort::ONNXRuntime> runtime_;

  // COND_SERIALIZABLE;
};

#endif