PeptDeepRescoring.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: Timo Sachsenberg $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/config.h>
#include <OpenMS/CONCEPT/ProgressLogger.h>
#include <OpenMS/DATASTRUCTURES/DefaultParamHandler.h>
#include <OpenMS/DATASTRUCTURES/ListUtils.h>
#include <OpenMS/KERNEL/StandardTypes.h>
 
#include <map>
#include <string>
#include <vector>
 
namespace OpenMS
{
  class PeptideIdentificationList;
  class ProteinIdentification;
  class PeptDeepMS2Inference;
  class PeptDeepRTInference;
 
  class OPENMS_DLLAPI PeptDeepRescoring :
    public DefaultParamHandler,
    public ProgressLogger
  {
  public:
    PeptDeepRescoring();
 
    ~PeptDeepRescoring() override;
 
    static StringList featureNames();
 
    static constexpr const char* NCE_META_VALUE = "peptdeep_nce";
 
    void annotate(const PeakMap& spectra,
                  std::vector<ProteinIdentification>& protein_ids,
                  PeptideIdentificationList& peptide_ids);
 
    double getUsedNCE() const;
 
    double getRTCalibrationError() const;
 
    Size getNumPSMs() const;
 
    Size getNumPredictedPSMs() const;
 
  protected:
    void updateMembers_() override;
 
    struct Row
    {
      Size pi;      
      Size hit;     
      Size len;     
      Size key;     
      double score; 
      bool calibration_candidate; 
    };
 
    void annotateRun_(const std::string& run_id,
                      const std::map<std::string, double>& collision_energies,
                      PeptideIdentificationList& peptide_ids,
                      const std::vector<Row>& rows,
                      const std::vector<Size>& run_rows,
                      const std::vector<std::string>& uniq_seq,
                      const std::vector<float>& uniq_charge,
                      bool higher_score_better,
                      PeptDeepMS2Inference& ms2_inference,
                      PeptDeepRTInference& rt_inference);
 
    std::string ms2_model_;
    std::string rt_model_;
    double nce_{-1.0};
    double nce_grid_halfwidth_{6.0};
    double nce_grid_step_{2.0};
    double nce_fallback_{30.0};
    Size nce_sample_size_{2000};
    std::string instrument_;
    double calibration_quantile_{0.5};
    std::string rt_model_type_{"b_spline"};
    Size rt_num_nodes_{5};
    int ms2_min_ordinal_{1};
    double ms2_strong_fraction_{0.05};
    Size batch_size_{500};
    Int threads_{4};
 
    double used_nce_{-1.0};
    double rt_calibration_error_{-1.0};
    Size num_psms_{0};
    Size num_predicted_psms_{0};
  };
 
} // namespace OpenMS