PeptDeepRTInference.h

Go to the documentation of this file.

// Copyright (c) 2002-present, OpenMS Team -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Timo Sachsenberg $
// $Authors: Satyam Yadav $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/config.h>
#include <OpenMS/ML/ONNX/ONNXPredictorBase.h>
#include <string>
#include <vector>
 
namespace OpenMS
{
    class OPENMS_DLLAPI PeptDeepRTInference
    {
    public:
        explicit PeptDeepRTInference(const std::string& model_path, int intra_op_threads = 4, size_t batch_size = 500);
 
        ~PeptDeepRTInference();
 
        std::vector<float> predictRT(const std::vector<std::string>& peptides);
 
    private:
        ONNXPredictorBase model_;
        size_t batch_size_;
    };
} // namespace OpenMS