dsa-decision-tree/DecisionNode.h at main · phamgialinhlx/dsa-decision-tree · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
/*
 * This file is part of dsa-decision-tree
 *
 * Developed for the DSA UET course.
 * This project was developed by Ba Luong and Gia Linh.
 */

#pragma once
#ifndef DECISION_NODE_H
#define DECISION_NODE_H

#include "Data.h"
#include "SplitData.h"
#include <sstream>
#include "Node.h"

class DecisionNode : virtual public Node
{
    private:
        int attribute;
        int compareValue;
        SplitData::SPLIT_VAL method;

    public:
        DecisionNode();
        ~DecisionNode();

        DecisionNode(int atr, int value, SplitData::SPLIT_VAL met);

        bool isTerminal();

        bool compare(Data *data);

        char getLabel();

        string toString();

        void setCode(int _code);

        string getExport();
};

#endif