GINI指数计算

GINI系数的计算

题目

1、计算整个训练集的基尼指数
2、计算训练集按照“性别”进行两路分割后的基尼指数
3、计算训练集按照“车型”进行三路分割后的基尼指数
4、（5） 计算训练集按照“衬衣尺码”进行四路分割后的基尼指数

思路

利用结构体和函数求解

代码

// A code block
#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

struct people           //建立结构体存放每一元组的数据
{
    int id;
    char sex;
    string car_type;
    string shirt_size;
    int pclass;
};


int main()
{
    int n;
    people *p;
    cin>>n;
    people person[n];                            //
    p=person;
    void whole(int n,people *p);                //计算整个训练集的GINI指数
    void ssex(int n,pe