PndKnnTrain.h

Go to the documentation of this file.

/* ************************************
 *  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