示例#1
0
int main() {
	cout << "!!!DataTeam project start!!!" << endl;

//	int totalTrainRecords = 878049;
	int totalTrainRecords = 100000;
	int totalTestRecords = 884261;
	string predictFieldName = "Category";
	string trainFileName = string("../DataTeam/files/train.csv");
	string testFileName = string("../DataTeam/files/test.csv");
	string trainNewFileName = string("../DataTeam/files/trainNew.csv");
	string submissionFileName = string("../DataTeam/files/submission.csv");

	//	Preparar el archivo para que funcione al pasarlo al NaiveBayes
	FileManager fileManager = FileManager(totalTrainRecords, trainFileName, trainNewFileName);
	fileManager.process();

	//	Pasarle el archivo con los datos de los registros como numericos
//	NaiveBayes naiveBayes = NaiveBayes(predictFieldName, fileManager.getSetterData());
//	naiveBayes.train(totalTrainRecords, trainNewFileName);
//	naiveBayes.test(totalTestRecords, testFileName, submissionFileName);

	KNN knn = KNN(50, fileManager.getSetterData());
	knn.aplicarKNN(trainNewFileName,testFileName,submissionFileName);

	cout << "Finish run app" << endl;
	return 0;
}