统计学习方法第四章，贝叶斯估计的实现

#include<iostream>
#include<vector>
using namespace std;
const int r=1;
 const int N=15;//训练样本个数
 const int K=2;//类标记数
const int M=3;//特征可能取值数
const int S=2;//特征维数
int array[3][N]={1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 1, 2, 2, 1, 1, 1, 2, 2, 3, 3, 3, 2, 2, 3, 3,-1, -1, 1, 1, -1, -1, -1, 1, 1, 1, 1, 1, 1, 1, -1};
float A[K];
float B[K][M];//记录第一维特征的取值与其相应的类的概率
float C[K][M];//记录第二维特征的取值与其相应的类的概率
int main(){
    int count1=0,count2=0;
    int count3[S][M];// y=-1
    int count4[S][M];//y=1;
    for(int i=0;i<S;i++)
        for(int j=0;j<M;j++)
        { 
            count3[i][j]=0;
            count4[i][j]=0;
        }
    for(int i=0;i<15;i++)
    {
        if(array[2][i]==-1){
            count1++;
            for(int j=0;j<2;j++){
                if(array[j][i]==1)
                {count3[j][0]++;}
                else
                    if(array[j][i]==2)
                    {count3[j][1]++;}
                    else
                        if(array[j][i]==3)
                            count3[j][2]++;
            }
        }
        else
            if(array[2][i]==1)
            {
                count2++;
                for(int j=0;j<2;j++)
                {
                    if(array[j][i]==1)
                        count4[j][0]++;
                    else 
                        if(array[j][i]==2)
                            count4[j][1]++;
                        else
                            if(array[j][i]==3)
                                count4[j][2]++;

                }
            }
    }
    cout<<count1<<endl;
    cout<<count2<<endl;
    for(int i=0;i<2;i++)
        for(int j=0;j<3;j++)
            cout<<count3[i][j];
    cout<<endl;
    A[0]=float(count1+r)/(N+K);
    A[1]=float(count2+r)/(N+K);
    cout<<A[0]<<endl;
    cout<<A[1]<<endl;
    for(int j=0;j<M;j++)
    {
        B[0][j]=float(count3[0][j]+r)/(count1+M*r);
        C[0][j]=float(count3[1][j]+r)/(count1+M*r);

    }
    for(int j=0;j<M;j++)
    {
        B[1][j]=float(count4[0][j])/count2;
        C[1][j]=float(count4[1][j])/count2;
    }
    int x1=0;
    cout<<"enter x(1):from 1 to 3:\n";
    int x2=0;
    char c;
    cin>>x1;
    cout<<"enter x(2):from s m l:\n";
    cin>>c;
    switch(c){
        case 's':x2=1;break;
        case 'm':x2=2;break;
        case 'l':x2=3;break;

    }
    cout<<(A[0]*B[0][x1-1]*C[0][x2-1])<<endl;
   // cout<<A[0]<<endl;
   // cout<<B[0][x1-1]<<endl;
   // cout<<C[0][x2-1]<<endl;
   // cout<<(A[1]*B[1][x1-1]*C[1][x2-1])<<endl;
    if((A[0]*B[0][x1-1]*C[0][x2-1])>(A[1]*B[1][x1-1]*C[1][x2-1]))
        cout<<"y= -1"<<endl;
    else
        cout<<"y= 1 "<<endl;
    return 0;
}