Gradient boosted trees as an NNModuleSpec #
The model spec defines the ensemble prediction function. This file adds the NNModuleSpec wrapper
for composition and export.
def
Spec.GradientBoostedTreesModuleSpec
{α : Type}
[Context α]
{nTrees maxDepth nFeatures : ℕ}
(model : GradientBoostedTreesSpec α nTrees maxDepth)
:
ModSpec.NNModuleSpec α (Shape.dim nFeatures Shape.scalar) Shape.scalar
Gradient boosted trees as an NNModuleSpec.