knn.h

来自「knn和Native Bayes算法实现,两个实现在一起,是数据挖掘和机器学习中」· C头文件 代码 · 共 28 行

#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