MarkTensenSgt

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

1

Can I find documentation somewhere? I've foudn the [colab notebook ](https://colab.research.google.com/github/alvinwan/neural-backed-decision-trees/blob/master/examples/load_pretrained_nbdts.ipynb#scrollTo=2ixOlZRnOnkx ) useful for the single prediction, but I can't get the different labels of a forward with decision tree.