rule.h

关联分类算法采用贪心算法发现高质量分类规则

// RULE.h: interface for the RULE class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_PREDICATE_H__9DE568BD_B5B2_497D_AF55_293E0C57B1C4__INCLUDED_)
#define AFX_PREDICATE_H__9DE568BD_B5B2_497D_AF55_293E0C57B1C4__INCLUDED_

#include <list>

class RULE  
{
public:
	RULE();
	RULE(const RULE& rule);
	virtual ~RULE();

	void	operator=(const RULE& rule);
	bool	operator==(const RULE& rule) const;

	void	SetClass(int iClass){m_Class=iClass;}
	int		GetClass() const {return m_Class;}
	int		GetSize() const {return m_Literals.size();}
	LITERAL	GetLiteral(int i) const;

	void	SetInfoGain(double gain){m_InfoGain=gain;}
	double	GetInfoGain() const {return m_InfoGain;}
	void	SetSupport(double sup){m_Support=sup;}
	double	GetSupport() const {return m_Support;}
	void	SetConfidence(double conf){m_Confidence=conf;}
	double	GetConfidence() const {return m_Confidence;}

	void	AddLiteral(const LITERAL& lit);
	bool	Satisfy(const TUPLE& tuple) const;

	void	WriteTo(FILE* stream) const;
	bool	ReadFrom(FILE* stream);

protected:	
	std::list<LITERAL> m_Literals;
	int		m_Class;

	double	m_InfoGain;
	double	m_Support;
	double	m_Confidence;
};


#endif // !defined(AFX_PREDICATE_H__9DE568BD_B5B2_497D_AF55_293E0C57B1C4__INCLUDED_)