Method predict

src/ml/onnx/onnx_classifier.cpp:65–391

int OnnxClassifier::predict (double *data, int data_len, double *output, int *output_len)
{
    int res = (int)BrainFlowExitCodes::STATUS_OK;
    if (ort == NULL)
    {
        return (int)BrainFlowExitCodes::CLASSIFIER_IS_NOT_PREPARED_ERROR;
    }
    if ((data == NULL) || (data_len < 1) || (output == NULL) || (output_len == NULL))
    {
        safe_logger (spdlog::level::err, "invalid input arguments");
        return (int)BrainFlowExitCodes::INVALID_ARGUMENTS_ERROR;
    }

    // todo add support for ints and float16
    float *float_data = NULL;
    if (input_type == ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT)
    {
        float_data = new float[data_len];
        for (int i = 0; i < data_len; i++)
        {
            float_data[i] = (float)data[i];
        }
    }
    else if (input_type == ONNX_TENSOR_ELEMENT_DATA_TYPE_DOUBLE)
    {
        // no need to convert
    }
    else
    {
        safe_logger (
            spdlog::level::err, "only float and double input types are currently supported");
        return (int)BrainFlowExitCodes::INVALID_ARGUMENTS_ERROR;
    }

    // create input tensor object from data values
    OrtMemoryInfo *memory_info = NULL;
    OrtValue *input_tensor = NULL;
    OrtValue *output_tensor = NULL;
    OrtStatus *onnx_status =
        ort->CreateCpuMemoryInfo (OrtArenaAllocator, OrtMemTypeDefault, &memory_info);
    if (onnx_status != NULL)
    {
        const char *msg = ort->GetErrorMessage (onnx_status);
        safe_logger (spdlog::level::err, "CreateCpuMemoryInfo failed: {}", msg);
        ort->ReleaseStatus (onnx_status);
        res = (int)BrainFlowExitCodes::GENERAL_ERROR;
    }
    else if (memory_info == NULL)
    {
        safe_logger (spdlog::level::err, "CreateCpuMemoryInfo failed");
        res = (int)BrainFlowExitCodes::GENERAL_ERROR;
    }
    if (res == (int)BrainFlowExitCodes::STATUS_OK)
    {
        if (input_type == ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT)
        {
            onnx_status = ort->CreateTensorWithDataAsOrtValue (memory_info, float_data,
                data_len * sizeof (float), input_node_dims.data (), input_node_dims.size (),