#ifndef CLASSIFIER__H
#define CLASSIFIER__H

#include <string>
#include <torch/torch.h>
#include "TransitionSet.hpp"

class Classifier
{
  private :

  std::string name;
  std::unique_ptr<TransitionSet> transitionSet;

  public :

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

#endif