neural-backed-decision-trees icon indicating copy to clipboard operation
neural-backed-decision-trees copied to clipboard

How do I unpack the output of model.forward_with_decisions(x) ?

Open MarkTensenSgt opened this issue 4 years ago • 1 comments

Can I find documentation somewhere? I've foudn the colab notebook useful for the single prediction, but I can't get the different labels of a forward with decision tree.

MarkTensenSgt avatar May 05 '21 14:05 MarkTensenSgt

Hey @MarkTensenSgt,

Thanks for posting! Here's the structure for each decision in that output https://github.com/alvinwan/neural-backed-decision-trees/blob/fe22aefe99afce813d3afb4a5976946e33e844f0/nbdt/model.py#L175. There isn't any formal documentation unfortunately.

alvinwan avatar Jul 24 '21 06:07 alvinwan