感知机 随机梯度下降代码

#include <map>
#include <queue>
#include <string>
#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
const int siz=10005;
int N,D,Yita;                                   //样本容量，属性个数，步长
int y[siz];                                     //标签
vector<int> x[siz];                             //样本
vector<int> w;
int b;                                          //参数w,b
void init_in(){
    int i,j,u;
    freopen("in.txt","r",stdin);                //数据就是<统计学习方法>例2.1
    scanf("%d %d",&D,&N);
    for(i=0;i<N;i++){
         for(j=0;j<D;j++){
            scanf("%d",&u);
            x[i].push_back(u);
        }
        scanf("%d",&y[i]);
    }
    w.clear();
    for(i=0;i<D;i++)
    w.push_back(0);
    b=0,Yita=1;                                 //w,b都初始为0，步长初始为1
}
int SGD_judge(int id){                          //通过yi*(w*xi+b)的正负判断数据分类正误
    int i,u=0;
    for(i=0;i<D;i++)
    u+=(x[id][i]*w[i]);
    u+=b;
    if(y[id]*u<=0)
    return 0;
    return 1;
}
void SGD(){
    int i,j,u;
    while(1){
        u=0;
        for(i=0;i<N;i++){                       //随机梯度下降直到所有数据都被正确分开
            if(SGD_judge(i)==0){                //w+Yita*xi*yi=>w
                u=1;                            //b+Yita*yi=>b
                for(j=0;j<D;j++)
                w[j]+=Yita*x[i][j]*y[i];
                b+=Yita*y[i];
                break;
            }
        }
        if(u==0)
        break;
    }
}
void SGD_output(){
    int i;
    for(i=0;i<D;i++){
        if(w[i]>0&&i!=0)
        printf("+");
        if(w[i]==1)
        printf("x%d",i+1);
        else
        printf("%dx%d",w[i],i+1);
    }
    if(b>0)
    printf("+");
    printf("%d=0\n",b);
}
int main(){
    init_in();
    SGD();
    SGD_output();
    return 0;
}