random_forest.go

func (r *randomForestGenerator) ConstructForest(e Examples) *pb.Forest {
	result := &pb.Forest{
		Trees:     make([]*pb.TreeNode, int(r.forestConfig.GetNumWeakLearners())),
		Rescaling: pb.Rescaling_AVERAGING.Enum(),
	}

	wg := sync.WaitGroup{}
	for i := 0; i < int(r.forestConfig.GetNumWeakLearners()); i++ {
		wg.Add(1)
		go func(i int) {
			result.Trees[i] = r.constructRandomTree(e)
			wg.Done()
		}(i)
	}
	wg.Wait()
	return result
}