PeptDeepMS2Inference.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 <string>
#include <vector>
#include <cstdint>
#include <OpenMS/config.h>
#include <OpenMS/ML/ONNX/ONNXPredictorBase.h>
 
namespace OpenMS {
 
class OPENMS_DLLAPI PeptDeepMS2Inference {
public:
    explicit PeptDeepMS2Inference(const std::string& model_path, int intra_op_threads = 4, size_t batch_size = 500);
 
    ~PeptDeepMS2Inference();
 
    std::vector<std::vector<float>> predictMS2(
        const std::vector<std::string>& peptides,
        const std::vector<float>& charges,
        const std::vector<float>& nces,
        const std::vector<int64_t>& instrument_indices);
 
private:
    ONNXPredictorBase model_;
    size_t batch_size_;
};
 
} // namespace OpenMS