C++实现KNN算法

最新推荐文章于 2025-03-13 10:24:01 发布

逸然逸生

最新推荐文章于 2025-03-13 10:24:01 发布

阅读量1.3k

点赞数 2

分类专栏： C++ 机器学习文章标签： C++ KNN 深度学习机器学习

本文链接：https://blog.csdn.net/qq_38502736/article/details/107232757

版权

C++实现KNN算法、

/*
 * @Description: C++实现KNN算法
 * @Author: szq
 * @Github: https://github.com/MrQqqq
 * @Date: 2020-07-08 19:13:25
 * @LastEditors: szq
 * @LastEditTime: 2020-07-09 16:50:55
 * @FilePath: \cpp\src\KNN\KNN.cpp
 */ 

#include<iostream>
#include<vector>
#include<fstream>
#include<random>
#include<time.h>
#include<map>
#include<algorithm>
using namespace std;

/**
 * @destription: 分割字符串
 * @param s:源字符
 * @param mode:分割的字符 
 * @return: 字符串分割后的字符串数组
 */
vector<string> split(string &s,char mode){
   
    vector<string> res;
    while(s.size() > 0){
   
        int index = s.find(mode);
        if(index != -1){
   
            res.push_back(s.substr(0,index+1));
            s = s.substr(index + 1);
        }
        else{
   
            res.push_back(s);
            break;
        }
        
    }
    return res;
}

/**
 * @destription: 获取文件的行数并将每一行的内容按','分割，保存起来
 * @param in:输入流
 * @param lines:保存每一行的结果
 * @return: 返回文件的行数
 */
int getFileRows(ifstream &in,vector<vector<string>> &lines){
   
    int rows = 0;
    char line[512];
    while(!in.eof()){
   
        in.getline(line,512,'\n');
        string src = string(line);
        lines.push_back(split(src,','));
        rows++;
    }
    return rows;
}

/**
 * @destription: 按一定的比例划分训练集和测试集
 * @param filepath:保存数据的文件地址
 * @param rate:训练集占中总数据的比例
 * @param trainingSet:训练集
 * @param testSet:测试集 
 * @return: 没有返回，结果都保存在对应的参数中
 */
void loadDataset(string &filepath,double &rate,vector<vector

最低0.47元/天解锁文章