#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
#include <string>
#include <iostream>
#include <cmath>
using namespace std;
#define clr(s,t) memset(s,t,sizeof(s))
#define threshhold 0.001
#define TRAIN 15
#define D 4
struct node{ //决策树的结点
int attr; //表示当前结点用哪个属性来区分,这里图方便设置成int的,实际上用string更好
string ans; //如果是叶子节点,这个域标示预测的结果
bool leaf; //指示当前结点是否为叶节点
map<string, int> br; //预测的时候用,string为属性值,int表示进入哪个儿子结点
vector<node> next; //儿子结点
}dt[10000];
typedef vector<pair<vector<string>,string>> DATA;//这是数据集类型
char fea[100][20]; //这里保存的是类标号
DATA data;
int used[D+1],num[1000],alloc = 1;//used表示属性有没有被用过,构建决策树的时候用
int len = -1; //表示类标号的数量,即fea数组中的元素个数
int find(const char* x){ //寻找类标号时候用
for(int i = 0;i&l
决策树——教材上的数据集
最新推荐文章于 2024-04-04 10:14:54 发布