decisiontree

Documentation

Overview

Package decisiontree contains utilities to handle decision trees.

Index

• Constants
• type Forest
  • func LoadForest(basePath string, numShards int, format string, numTrees int) (*Forest, error)
  • func (f *Forest) NumLeafs() int
  • func (f *Forest) NumNonLeafs() int
• type Node
  • func (n *Node) IsLeaf() bool
  • func (n *Node) NumLeafs() int
  • func (n *Node) NumNonLeafs() int
• type Tree

Constants

const DefaultNodeFilename = "nodes"

DefaultNodeFilename is the default filename to store nodes.

Types

type Forest

type Forest struct {
	Trees []*Tree
}

Forest is a collection of trees.

func LoadForest

func LoadForest(basePath string, numShards int, format string, numTrees int) (*Forest, error)

LoadForest loads a forest from disk.

func (*Forest) NumLeafs

func (f *Forest) NumLeafs() int

NumLeafs is the number of leafs in the forest.

func (*Forest) NumNonLeafs

func (f *Forest) NumNonLeafs() int

NumNonLeafs is the number of non-leaf nodes in the forest.

type Node

type Node struct {
	RawNode       *pb.Node
	PositiveChild *Node
	NegativeChild *Node
}

Node is a tree node.

func (*Node) IsLeaf

func (n *Node) IsLeaf() bool

IsLeaf tests if a node is a leaf.

func (*Node) NumLeafs

func (n *Node) NumLeafs() int

NumLeafs is the number of leafs in a sub-tree.

func (*Node) NumNonLeafs

func (n *Node) NumNonLeafs() int

NumNonLeafs is the number of non-leaf nodes in a sub-tree.

type Tree

type Tree struct {
	// Root node of the tree. nil if the tree is empty.
	Root *Node
}

Tree is a decision tree.