LeafDecisionNode.h

#ifndef GRAIL_LEAF_DECISION_NODE_H
#define GRAIL_LEAF_DECISION_NODE_H
 
#include "DecisionNode.h"
#include <memory>
#include <iostream>
 
namespace grail
{
 namespace simulation
 {
 template <class TDecisionType>
 class LeafDecisionNode : public DecisionNode<TDecisionType>
 {
 public:
 LeafDecisionNode(std::unique_ptr<const TDecisionType> decision);
 
 const TDecisionType* Predict(std::vector<float>&) const final
 {
 return decision.get();
 }
 
 void Print(std::unordered_map<int, std::string>&, const std::string& indent = "") const override
 {
 std::cout << indent << "Decision: " << *decision.get() << std::endl;
 }
 
 DecisionNodeType GetNodeType() const final
 {
 return DecisionNodeType::LEAF;
 }
 
 private:
 std::unique_ptr<const TDecisionType> decision;
 
 template <class T> friend struct IDecisionTreeSerializer;
 };
 
 template<class TDecisionType>
 inline LeafDecisionNode<TDecisionType>::LeafDecisionNode(std::unique_ptr<const TDecisionType> decision) :
 decision { std::move(decision)}
 
 {
 
 }
 }
}
 
#endif //GRAIL_LEAF_DECISION_NODE_H