DecisionTree.h

Go to the documentation of this file.

#ifndef DECISIONTREEH
#define DECISIONTREEH

#include "ExampleSpec.h"
#include "Example.h"
#include "../util/lists.h"
/*#include <stdio.h>*/


typedef enum { dtnLeaf, dtnContinuous, dtnDiscrete, dtnGrowing } NodeType;

typedef struct _DecisionTree_ {
   NodeType nodeType;

   ExampleSpecPtr spec;
   void *growingData;

   int splitAttribute;
   float splitThreshold; /* for continuous attributes */

   VoidAListPtr children;

   int myclass;

   float *classDistribution;
   float distributionSampleCount;
} DecisionTree, *DecisionTreePtr;

DecisionTreePtr DecisionTreeNew(ExampleSpecPtr spec);

void DecisionTreeFree(DecisionTreePtr dt);

DecisionTreePtr DecisionTreeClone(DecisionTreePtr dt);

int DecisionTreeIsLeaf(DecisionTreePtr dt);

int DecisionTreeIsTreeGrowing(DecisionTreePtr dt);

int DecisionTreeIsNodeGrowing(DecisionTreePtr dt);

int DecisionTreeGetClass(DecisionTreePtr dt);

void DecisionTreeSetClass(DecisionTreePtr dt, int theClass);

void DecisionTreeAddToClassDistribution(DecisionTreePtr dt, ExamplePtr e);

float DecisionTreeGetClassProb(DecisionTreePtr dt, int theClass);

void  DecisionTreeSetClassProb(DecisionTreePtr dt, int theClass, float prob);

float DecisionTreeGetClassDistributionSampleCount(DecisionTreePtr dt);

void DecisionTreeZeroClassDistribution(DecisionTreePtr dt);

void DecisionTreeSetTypeLeaf(DecisionTreePtr dt);

void DecisionTreeSetTypeGrowing(DecisionTreePtr dt);

void DecisionTreeSplitOnDiscreteAttribute(DecisionTreePtr dt,
                                                            int attNum);

void DecisionTreeSplitOnContinuousAttribute(DecisionTreePtr dt,
                                            int attNum, float threshold);

int DecisionTreeGetChildCount(DecisionTreePtr dt);

DecisionTreePtr DecisionTreeGetChild(DecisionTreePtr dt, int index);

DecisionTreePtr DecisionTreeOneStepClassify(DecisionTreePtr dt, ExamplePtr e);

int DecisionTreeClassify(DecisionTreePtr dt, ExamplePtr e);

void DecisionTreeGatherGrowingNodes(DecisionTreePtr dt, VoidAListPtr list);

void DecisionTreeGatherLeaves(DecisionTreePtr dt, VoidAListPtr list);

int  DecisionTreeCountNodes(DecisionTreePtr dt);

int DecisionTreeGetMostCommonClass(DecisionTreePtr dt);

void DecisionTreeSetGrowingData(DecisionTreePtr dt, void *data);


void *DecisionTreeGetGrowingData(DecisionTreePtr dt);


void DecisionTreePrint(DecisionTreePtr dt, FILE *out);


void DecisionTreePrintStats(DecisionTreePtr dt, FILE *out);


DecisionTreePtr DecisionTreeReadC45(FILE *in, ExampleSpecPtr spec);
DecisionTreePtr DecisionTreeReadC45Overprune(FILE *in, ExampleSpecPtr spec);

DecisionTreePtr DecisionTreeRead(FILE *in, ExampleSpecPtr spec);

void DecisionTreeWrite(DecisionTreePtr dt, FILE *out);

#endif /* DECISIONTREEH */

---

Generated for VFML by hosted by