func (s *Sample) ToString(includePrediction bool) []byte {
sb := util.StringBuilder{}
sb.Int(s.Label)
sb.Write(" ")
if includePrediction {
sb.Float(s.Prediction)
sb.Write(" ")
}
for _, feature := range s.Features {
sb.Int64(feature.Id)
sb.Write(":")
sb.Float(feature.Value)
sb.Write(" ")
}
return sb.Bytes()
}
func (t *Tree) ToString() []byte {
sb := util.StringBuilder{}
sb.Int(len(t.nodes))
sb.Write("\n")
for i, node := range t.nodes {
sb.Int(i)
sb.Write("\t")
sb.Int(node.left)
sb.Write("\t")
sb.Int(node.right)
sb.Write("\t")
sb.Int(node.depth)
sb.Write("\t")
sb.WriteBytes(node.prediction.ToString())
sb.Write("\t")
sb.Int(node.sample_count)
sb.Write("\t")
sb.Int64(node.feature_split.Id)
sb.Write("\t")
sb.Float(node.feature_split.Value)
sb.Write("\n")
}
return sb.Bytes()
}