io

Documentation

Overview

Package io contains utilities to load/save decision trees

Index

• Variables
• type Reader
  • func NewNodeReader(path string, numShards int, format string) (Reader, error)

Variables

var RegisteredFormats = make(map[string]func(path string) (Reader, error))

RegisteredFormats is the list of format readers.

Types

type Reader

type Reader interface {

	// Next returns the next raw Node protobuffer from the stream. Returns nil at the end of the
	// stream. "Close" should be called once the reading is done (even if file reaches end).
	Next() (*pb.Node, error)
	Close() error
}

Reader is a stream of nodes.

func NewNodeReader

func NewNodeReader(path string, numShards int, format string) (Reader, error)

NewNodeReader creates a new node reader from a sharded set of files.