pickle

type Build struct {
	Object interface{}
	// Args are usually represented like Tuple (arguments for custom __setstate__) or Dict (class members)
	Args interface{}
}

type Decoder struct {
	// contains filtered or unexported fields
}

type Dict map[string]interface{}

type Global struct {
	Module string
	Name   string
}

type List []interface{}

type None struct{}

type NumpyArrayRaw struct {
	Type  NumpyDType
	Shape []int
	// For plain data objects array data are stored in NumpyRawBytes
	Data NumpyRawBytes
	// But in case when data objects are python objects, they are parsed to []interface{}
	DataList []interface{}
}

type NumpyDType struct {
	LittleEndinan bool
	Type          string
}

type NumpyElementFunc func([]byte) error

type NumpyRawBytes []byte

type NumpyScalarRaw struct {
	Type NumpyDType
	Data NumpyRawBytes
}

type PythonClass interface {
	Reduce(reduce Reduce) error
	Build(build Build) error
}

type Reduce struct {
	Callable interface{}
	Args     Tuple
}

type SKlearnInitEstimator struct {
	Name  string
	Prior []float64
}

type SklearnDecisionTreeRegressor struct {
	Tree        SklearnTree
	NClasses    int
	MaxFeatures int
	NOutputs    int
}

type SklearnGradientBoosting struct {
	NClasses      int
	Classes       []int
	NEstimators   int
	MaxFeatures   int
	Estimators    []SklearnDecisionTreeRegressor
	LearningRate  float64
	InitEstimator SKlearnInitEstimator
}

type SklearnNode struct {
	LeftChild            int
	RightChild           int
	Feature              int
	Threshold            float64
	Impurity             float64
	NNodeSamples         int
	WeightedNNodeSamples float64
}

type SklearnTree struct {
	NOutputs int
	Classes  []int
	NNodes   int
	Nodes    []SklearnNode
	Values   []float64
}

type Tuple []interface{}

type Unicode []byte