rule.cpp

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

// RULE.cpp: implementation of the RULE class.
//
//////////////////////////////////////////////////////////////////////

#include "StdAfx.h"
#include "DESCRIPTION.h"
#include "TUPLE.h"
#include "LITERAL.h"
#include "RULE.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

RULE::RULE()
{
	m_Class=0;
	m_InfoGain=0.0;
	m_Support=0.0;
	m_Confidence=0.0;
}

RULE::~RULE()
{

}

void RULE::AddLiteral(const LITERAL& lit)
{
	LITERAL l=lit;
	m_Literals.push_back(l);
}

LITERAL	RULE::GetLiteral(int i) const
{
	if (i >= m_Literals.size()) {
		return LITERAL();
	} else {
		std::list<LITERAL>::const_iterator pos=m_Literals.begin();
		while(i>0) {
			pos ++;
			i --;
		}
		return *pos;
	}
}

bool RULE::Satisfy(const TUPLE& tuple) const
{
	if(m_Literals.empty()) return true;

	for(std::list<LITERAL>::const_iterator pos=m_Literals.begin();pos!=m_Literals.end();pos++)
	{
		if(!(pos->Satisfy(tuple))) return false;
	}
	return true;
}

RULE::RULE(const RULE& rule)
{
	(*this)=rule;
}

void RULE::operator=(const RULE& rule)
{
	m_Class=rule.GetClass();
	m_Literals.clear();
	for(std::list<LITERAL>::const_iterator pos=rule.m_Literals.begin();pos!=rule.m_Literals.end();pos++)
	{
		LITERAL lit=*pos;
		m_Literals.push_back(lit);
	}
	m_InfoGain=rule.m_InfoGain;
	m_Support=rule.m_Support;
	m_Confidence=rule.m_Confidence;
}

bool RULE::operator==(const RULE& rule) const
{
	return m_Literals == rule.m_Literals;
}

void RULE::WriteTo(FILE* stream) const
{
	fprintf(stream,"CLASS=%d ( %.3f %d %.2f ) : ",m_Class,m_InfoGain,(int)(m_Support+0.5),m_Confidence);
	for(std::list<LITERAL>::const_iterator pos=m_Literals.begin();pos!=m_Literals.end();pos++)
	{
		pos->WriteTo(stream);
		fprintf(stream," ");
	}
	fprintf(stream,"\n");
}

bool RULE::ReadFrom(FILE* stream)
{
	char buf1[200],buf2[200];
	int sup;
	int res=fscanf(stream,"CLASS=%d ( %s %d %s ) : ",&m_Class,buf1,&sup,buf2);
	if(res!=4) return false;
	m_InfoGain=atof(buf1);
	m_Support=sup;
	m_Confidence=atof(buf2);

	while(1)
	{
		LITERAL lit;
		if(lit.ReadFrom(stream))
		{
			AddLiteral(lit);
			fscanf(stream," ");
		}
		else break;
	}

	fscanf(stream,"\n");
	return true;
}