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; }