00001 #ifndef C_KDTREES__H 00002 #define C_KDTREES__H 00003 00004 00005 #include "ctrees.h" 00006 00007 class CDataSet; 00008 00009 class CKDTreeMedianSplittingFactory : public CSplittingConditionFactory 00010 { 00011 protected: 00012 int n_min; 00013 CDataSet *inputData; 00014 public: 00015 CKDTreeMedianSplittingFactory(CDataSet *inputSet, int n_min); 00016 virtual ~CKDTreeMedianSplittingFactory(); 00017 00018 virtual CSplittingCondition *createSplittingCondition(DataSubset *dataSubset); 00019 00020 virtual bool isLeaf(DataSubset *dataSubset); 00021 }; 00022 00023 class CKDTree : public CTree<DataSubset *> 00024 { 00025 protected: 00026 CSplittingConditionFactory *splittingFactory; 00027 public: 00028 CKDTree(CDataSet *dataSet, int n_min); 00029 virtual ~CKDTree(); 00030 00031 virtual void addNewInput(int index); 00032 }; 00033 00034 #endif
