neural-backed-decision-trees
neural-backed-decision-trees copied to clipboard
How do I unpack the output of model.forward_with_decisions(x) ?
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.
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.