knn.h

【机器学习】文本分类源代码 【机器学习】文本分类源代码

#ifndef __KNN_H_
#define __KNN_H_

#pragma warning( disable : 4786 )

#include "doclist.h"


class CKNN
{
public:
	unsigned int k;

	CKNN();
	~CKNN();
	//void Train(DOC *docs, vector<set<int> >& target, long totdoc);
	void Train( CDocList& DocList );
	//void TestDocs(DOC *docs, vector<set<int> >& target, long totdoc);
	float CalcSim(WORDITEM *p, WORDITEM *q);
	int ClassifyDocs( DOC& test_doc );
	void ClassifyDocs( string sVectorFile, string sResultFile );
private:
	DOC *docs_train;
	vector<set<int> > vTargetTrain;
	long lTrainTotal;
};

#endif