tree.h

自主式决策树学习的程序源码

// Tree.h: interface for the CTree class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_TREE_H__047FE5D9_FD20_49BB_B4B0_2F6FCE2170C7__INCLUDED_)
#define AFX_TREE_H__047FE5D9_FD20_49BB_B4B0_2F6FCE2170C7__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

struct TreeNode
{
	CString  SplitVal;//记录父分裂结点生成该结点所用的属性值
	CString SplitAtt;//该结点继续分裂时所用的分裂属性名
	int ChiCount;//孩子结点的个数
	TreeNode ** ChiNode;//孩子结点
	TreeNode * ParNode;//父结点
	char * AttArray;//当前条件属性集
	int  AttCount;//当前条件属性个数，另一个用途是当属性个数大于最初个数时，表示第一类叶结点
	CString strWh;//SQL语句中用于确定当前的记录集
	float  CerRatio;//非叶结点的整体确定性
    CString ClassName;//叶结点的分类名
	float   LeaveCer;//叶结点的确定性
	float   supRatio;//叶结点的支持度
    CPtrList * nodeList;//指向该节点对应的list，用于以后释放空间
	
};

class CTree  
{
public:
	TreeNode *TreeBoot;
public:
	CTree();
	virtual ~CTree();
    void CreatTree(TreeNode * ptn);//建树
	void delTreeNode(TreeNode * ptn);//删除树该节点
	void FreeList(CPtrList * list);//删除链
	void arrSub(char * sArr,int size,char *dArr, char ch);//获取新数组
    void rules(TreeNode * ptn);//测试时用于输出规则集
	void reduceRules(TreeNode * ptn);//生成规则集
	void delTree(TreeNode *ptn);//删除以该结点为根的树
};

#endif // !defined(AFX_TREE_H__047FE5D9_FD20_49BB_B4B0_2F6FCE2170C7__INCLUDED_)