Skip to content
Snippets Groups Projects
Classifier.hpp 391 B
Newer Older
#ifndef CLASSIFIER__H
#define CLASSIFIER__H

#include <string>
#include "TransitionSet.hpp"
Franck Dary's avatar
Franck Dary committed
#include "MLP.hpp"

class Classifier
{
  private :

  std::string name;
  std::unique_ptr<TransitionSet> transitionSet;
Franck Dary's avatar
Franck Dary committed
  MLP nn{nullptr};

  public :

  Classifier(const std::string & name, const std::string & topology, const std::string & tsFile);
  TransitionSet & getTransitionSet();