PndStdKnnClassify.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

/* ***************************************
 * KNN based classifier.                 *
 * Author: M.Babai@rug.nl                *
 * Edited: E.A.Dijck@student.rug.nl      *
 * Version:                              *
 * license:                              *
 * ***************************************
 */
/*
 * Based on the straight KNN algorithm
 * Using a linear search data structure.
 */
//#pragma once
#ifndef PND_STD_KNN_CLASSIFY_H
#define PND_STD_KNN_CLASSIFY_H

// Local includes
#include "PndMvaClassifier.h"

/*
 * KNN based classification alg. implementation.
 */
class PndStdKnnClassify : public PndMvaClassifier {
 public:
  explicit PndStdKnnClassify(std::string const &inputFile, std::vector<std::string> const &classNames, std::vector<std::string> const &varNames);

  virtual ~PndStdKnnClassify();

  void GetMvaValues(std::vector<float> eventData, std::map<std::string, float> &result);
  std::string *Classify(std::vector<float> EvtData);

  inline void SetKNN(unsigned int const nNeighbours);

  inline unsigned int GetKnn() const;

  void Initialize();

 private:
  PndStdKnnClassify(const PndStdKnnClassify &other);
  PndStdKnnClassify &operator=(const PndStdKnnClassify &other);

  unsigned int m_Knn;

  std::vector<PndMvaDistObj> m_distances;
}; // End interface.
// ____________________________ Inlines ________________
inline unsigned int PndStdKnnClassify::GetKnn() const
{
  return m_Knn;
};

inline void PndStdKnnClassify::SetKNN(unsigned int const nNeighbours)
{
  m_Knn = nNeighbours;
};
#endif