tree_node.h

Go to the documentation of this file.

#ifndef _TREE_NODE_H_
#include "librf/types.h"
#include <fstream>
using namespace std;

namespace librf {

typedef enum {EMPTY, BUILD_ME, TERMINAL, SPLIT} NodeStatusType;

struct tree_node {
  tree_node() :status(EMPTY),
               label(99),
               attr(99),
               start(99),
               size(99),
               split_point(-999.0),
               entropy(-9999.0){}
  NodeStatusType status;
  uchar label;
  uint16 attr;
  uint16 start;
  uint16 size;
  float entropy;
  float split_point;
  uchar depth;

  void write(ostream& o) const;
  void read(istream& i);
};
} //namespace
#endif

---