neural-backed-decision-trees

neural-backed-decision-trees copied to clipboard

Can you answer how Zero-Shot Superclass Generalization is calculated?

dinah-wang

Hi, Great project and article, congratulations. I'm wondering if, once the training is finished, the final NBDT can be fully exported as a JIT file? It seems that everything is...

Optimox

study1157

I mean apply NBDT with soft loss.., you know the inner node weight can't be update with the soft loss in binary classfication due to the special inner probability computing...

study1157

How to train with new dataset

1

Hi, I'm trying to use gen_train_eval_nopretrained.sh to train with a new dataset I implemented. However, in main.py there is this line of code tree = Tree.create_from_args(args, classes=trainset.classes) the error I...

XAVILLA

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.

MarkTensenSgt

Windows support?

1

Can it run under Windows system， and how?

wangdemo

Regression problems

1

Is there any example for a regression problem?

kurucan

error when use 'nbdt-hierarchy'

4

when run command 'nbdt-hierarchy' in ubuntu 18.04 **nbdt-hierarchy --checkpoint=wrn28_10_cifar10 --dataset=CIFAR10 nbdt-hierarchy --vis-sublabels --vis-zoom=1.25 --dataset=CIFAR10 --arch=wrn28_10_cifar10** Here is ModuleNotFoundError: No module named 'nbdt.thirdparty', I wonder what happened

zhuyi3625

Probability of outputs not ensure

1

it has different outputs when I input the same picture into the model which is trained by myself.However when use the pretrained '.pth' model,the output is right,and the probability tensor...

zhuyi3625