ID3.h

#ifndef __ClASSIFY_TREE_H__
#define __CLASSIFY_TREE_H__

#include "BaseTree.h"
#include "Feature.h"

namespace mla {
namespace tree {

class ID3 : public mla::basic::Tree<float> {
public:

    ID3() {} 
    
    ID3(int32_t maxNodeCnt, 
            int32_t maxDepth, 
            bool isMultiThreadOn,
            int32_t labelCnt,
            bool ensemble) : mla::basic::Tree<float>(maxNodeCnt, maxDepth, isMultiThreadOn, labelCnt, ensemble) {}
    
    void optSplitPos(int &nOptFeatureIndex,
                float &nOptFeatureVal,
                std::vector<int32_t> &vCurrentIndex,
                std::vector<int32_t> &vFeatureIndex); 

    void optSplitPosMultiThread(int &nOptFeatureIndex,
            float &nOptFeatureVal,
            std::vector<int32_t> &vCurrentIndex,
            std::vector<int32_t> &vFeatureIndex);

    void splitData(struct mla::basic::Node<float>* &node,
        const int &nOptFeatureIndex,
        const float &fOptFeatureVal,
        const std::vector<int32_t> &vTempCurrentIndex,
        std::vector<int32_t> &vLeftIndex,
        std::vector<int32_t> &vRightIndex);

    friend void* selectFeatureFuncC(void* param);

    float predict(const std::vector<float> &testFeatureX);
};
}
}

#endif