https://github.com/alvinwan/neural-backed-decision-trees

sort by:
Revision Author Date Message Commit Date
189b191 version bump 30 March 2020, 23:47:49 UTC
4fb1638 hierarchy download=True 30 March 2020, 23:45:14 UTC
452f987 updated vis 30 March 2020, 23:41:15 UTC
dba9b53 addedd citation, results, authors 30 March 2020, 23:22:41 UTC
4545b23 fixed models used for inference 30 March 2020, 11:15:09 UTC
bf721d3 move zebra to line of pics 30 March 2020, 11:14:14 UTC
7722bb8 update outputs 30 March 2020, 11:13:02 UTC
1c3a8ef omg i frogot .eval() 30 March 2020, 11:03:22 UTC
b4e030e ixes 30 March 2020, 10:48:03 UTC
be05081 module. heckinng 30 March 2020, 10:35:58 UTC
8c3308c updated acc 30 March 2020, 10:08:18 UTC
a0b75c3 ability to color specific noddes in vis 30 March 2020, 10:05:25 UTC
e3032b8 added wordnet labels to induced hierarchy where possible 30 March 2020, 08:56:22 UTC
9047778 coerce all state dicts 30 March 2020, 08:56:09 UTC
227adf8 whoops this commit and previous added hard NBDT probs + wordnet pretrained model 30 March 2020, 08:31:09 UTC
c5ad0c5 added wordnet model 30 March 2020, 08:29:09 UTC
1cc6092 change text color too 30 March 2020, 00:36:46 UTC
6b772a7 update fname depending on path to leaf 29 March 2020, 09:11:46 UTC
bc4491c dark mode 29 March 2020, 08:41:24 UTC
0b993f2 support colored edges for path coloring 29 March 2020, 08:19:29 UTC
bf5bf1b added blue-green gradient + ability to color path to leaf 29 March 2020, 07:22:40 UTC
6f92155 configure vis height + fix wordnet attempt to import 28 March 2020, 01:12:48 UTC
e340853 update nums, remove extra notebook 28 March 2020, 01:10:41 UTC
09ec076 updated links 27 March 2020, 11:02:20 UTC
24deb97 slightly more proper api 27 March 2020, 08:47:32 UTC
ef8661f *finally* got a working version 🤬😡 26 March 2020, 06:54:21 UTC
74400c4 minimal app functionality (inferrence on url via form, uploaded photo, or url via queryparam (click on photo) 26 March 2020, 05:13:42 UTC
b53d6b4 deploying via aws elastic beanstalk 26 March 2020, 04:15:18 UTC
d09f706 try again with correct manifest 26 March 2020, 00:01:59 UTC
1b36b6a version bump, license, docstring 25 March 2020, 23:38:58 UTC
4dc43d8 link to web demo 25 March 2020, 23:09:22 UTC
38b2812 example app 25 March 2020, 22:45:51 UTC
0f23c98 added onnx export (but check failing on models) + load_state_dict fixed for old checkpoints 25 March 2020, 22:21:52 UTC
48280da added google colab link 25 March 2020, 20:46:53 UTC
6369053 fixed soft inference + nbdt script 25 March 2020, 20:41:20 UTC
3810d21 fix cat 🐈 25 March 2020, 19:19:19 UTC
6f7443f added git clone note 25 March 2020, 17:59:47 UTC
9454256 move eye tensor 25 March 2020, 17:37:16 UTC
0da19b4 fixed hard inference 25 March 2020, 16:55:35 UTC
66a55d6 model returns logits, not prediction 25 March 2020, 16:43:04 UTC
95b6953 unify dataset name in example 25 March 2020, 12:09:57 UTC
e7a51fa Added effnetb7 hierarchy 25 March 2020, 11:19:41 UTC
7514d16 all utils will insntall wordnet if ont founnd 25 March 2020, 11:11:23 UTC
a56bd7f added wnid instructions for util 25 March 2020, 10:59:46 UTC
851b8b5 impute with fake wnids in nbdt-widds 25 March 2020, 10:52:14 UTC
12e0d03 fixed mai 25 March 2020, 10:40:38 UTC
7f3807e nbdt-wnids supports all datasets 25 March 2020, 10:32:53 UTC
49d05ad renamed iduced_checkpoinnt to checkpoint 25 March 2020, 10:12:20 UTC
b83da0d unify all model specs to --arch + remove --single-path 25 March 2020, 09:53:54 UTC
54f05cb remove badges 25 March 2020, 08:35:05 UTC
d17e349 Update README.md 25 March 2020, 08:33:39 UTC
6645935 added flag to automatically add pictures to tree vis 25 March 2020, 08:04:52 UTC
472cf29 Add cli support for forcing text to left 25 March 2020, 03:40:56 UTC
9970eca placeholder badges 24 March 2020, 10:31:07 UTC
fa4a241 added instructions for reproducing vis/figs 24 March 2020, 09:11:47 UTC
b001e27 Addd more resnet pretrained models 24 March 2020, 09:11:17 UTC
04477bb ability to color all nodes gray + default color leaves blue 24 March 2020, 08:38:00 UTC
9a2bd94 cchange font, fix resizing padding, noe sizes 24 March 2020, 07:29:33 UTC
29f4392 more configuration options for hierarchy vis + closer to figure style 24 March 2020, 06:19:22 UTC
9d7da91 combine wnid_to_synset, synset_to_name + fix ImageNet, WordNet capitalization convention 24 March 2020, 01:28:11 UTC
e200c60 fix embedded decision rules oops 24 March 2020, 01:03:38 UTC
6bcff6d addedd wordnet hierarchies, made single path wordnet the default 24 March 2020, 00:16:01 UTC
3429f9f fixed soft decision, new path format 24 March 2020, 00:10:37 UTC
0d580d0 fixed tinyimagenet dataloader download 23 March 2020, 23:23:17 UTC
1f708e7 fixed default xent loss kwarg 23 March 2020, 23:02:12 UTC
ab6359f make hierarchy optional 23 March 2020, 22:57:07 UTC
7d4ee45 added method + output decisions in script 23 March 2020, 21:13:34 UTC
92652df added moddel urls 23 March 2020, 09:12:31 UTC
492007d loading model example, less clutter in 3-step 23 March 2020, 07:25:30 UTC
c8926e0 siompler sample script 23 March 2020, 07:17:32 UTC
71555e3 updated r18 hierarchies 23 March 2020, 06:43:47 UTC
1a0be99 fixed hard inference 23 March 2020, 06:30:57 UTC
bfde8f3 pass criterion 23 March 2020, 06:13:52 UTC
e6dc5b3 populate classes arg in main 23 March 2020, 06:12:44 UTC
13f21ef whoopsx2 23 March 2020, 05:46:20 UTC
04f05e8 whoops, import 23 March 2020, 05:45:15 UTC
866bbb1 fixed analyzer construtor 23 March 2020, 05:43:57 UTC
30c84c3 support passing pretrained model to generate_hierarchy 23 March 2020, 05:11:15 UTC
c834e88 args-less generate_hierarchy func, moved bin to nbdt 23 March 2020, 05:07:50 UTC
9cd5de9 removed unused models 23 March 2020, 04:48:22 UTC
1d58119 added support for --hierarchy flag instead of --path-graph 23 March 2020, 04:43:25 UTC
9e50d26 🌞 23 March 2020, 04:35:19 UTC
126a221 added bash scripts for modification, both for networks with pretrained and those without pretrained ccheckpoints 23 March 2020, 04:32:05 UTC
6208f3e way fewer scripts + fleshed out dev section 23 March 2020, 04:12:31 UTC
762db54 simpler training/eval 23 March 2020, 03:36:35 UTC
c06de4b nbdt-hierarchy now supports model name 23 March 2020, 00:52:42 UTC
4a93c4b always log graph hierarchy 23 March 2020, 00:06:37 UTC
c34c659 added nbdt eval script 22 March 2020, 15:06:05 UTC
729e2dc made generate_hiehrarchy and generate_wnids globally accessible scripts 22 March 2020, 14:43:39 UTC
c8b20bb combined generate_vis and gegnerate_hierarchies 22 March 2020, 14:17:22 UTC
42660b5 combined test + generate hierarchies 22 March 2020, 14:06:31 UTC
cf8efe6 oops, 'net' 22 March 2020, 13:37:21 UTC
1ff514a support pretrained kwarg for nbdt models 22 March 2020, 13:37:21 UTC
ca79168 added support for pretrained models 22 March 2020, 13:37:21 UTC
79c5d2d list all imports 22 March 2020, 11:53:19 UTC
a1b3841 use wrn for example isntead 22 March 2020, 11:45:48 UTC
23120df re-added cifar10/cifar100/tinyimagennet friendly resnets 22 March 2020, 11:42:23 UTC
c2c695e added test for nbdt model -> nbdt loss check (fixed analysis) 22 March 2020, 11:29:36 UTC
fc3b206 update with new interface 22 March 2020, 10:53:13 UTC
0ff05c4 updated scripts to use torchvision model name 22 March 2020, 10:48:02 UTC
back to top