PndKnnTrain.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".              *
//****************************************************************************

/* ************************************
 *  Author: M. Babai (M.Babai@rug.nl) *
 *                                    *
 *  pid classifier                    *
 *                                    *
 * Modified:                          *
 *                                    *
 * ************************************/
//#pragma once
#ifndef PND_KNN_TRAIN_H
#define PND_KNN_TRAIN_H

#include "PndMvaTrainer.h"

class PndKnnTrain : public PndMvaTrainer {
 public:
  explicit PndKnnTrain(std::string const &InputFile, std::vector<std::string> const &ClassNames, std::vector<std::string> const &VarNames, bool trim = true);
  virtual ~PndKnnTrain();

  void Train();

  void storeWeights();

 private:
  PndKnnTrain(PndKnnTrain const &other);
  PndKnnTrain &operator=(PndKnnTrain const &other);
};
// end of interface definition
#endif