Testing addition of feature metadata to onnx model

import os

import numpy as np

import onnxruntime as rt

import onnx as ox

import logging

import time

LOGGER = logging.getLogger(__name__)

current_dir = os.path.dirname(os.path.abspath(__file__))

MODEL_FILE = os.path.join(current_dir, "ml_model/crypto_5g_rf_spider_features.onnx")

MODEL_FILE = os.path.join(current_dir, "ml_model/crypto_auto_features.onnx")

class l3_centralizedattackdetectorServiceServicerImpl(L3CentralizedattackdetectorServicer):

    def __init__(self):

        LOGGER.info("Creating Centralized Attack Detector Service")

        onnx_model = ox.load_model(MODEL_FILE)

        meta = onnx_model.metadata_props.add()

        meta.key = "key"

        meta.value = "value"

        LOGGER.debug(onnx_model.metadata_props[0])

        self.inference_values = []

        self.inference_results = []

        self.model = rt.InferenceSession(MODEL_FILE)

        '''self.model._model_meta = metadata_proto

        meta = self.model.get_modelmeta()

        LOGGER.debug(meta.description)

        time.sleep(10)'''

        LOGGER.debug(meta.custom_metadata_map)

        LOGGER.debug("Prueba onnx")

        self.input_name = self.model.get_inputs()[0].name

        self.label_name = self.model.get_outputs()[0].name