├── README.md ├── overview.pdf ├── overview.png ├── r.png ├── treeMN-non-hierachical ├── load_data.py ├── main.py ├── params.py ├── predict.py └── treeMN.py └── treeMN ├── load_data.py ├── main.py ├── params.py └── treeMN.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/README.md -------------------------------------------------------------------------------- /overview.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/overview.pdf -------------------------------------------------------------------------------- /overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/overview.png -------------------------------------------------------------------------------- /r.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/r.png -------------------------------------------------------------------------------- /treeMN-non-hierachical/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN-non-hierachical/load_data.py -------------------------------------------------------------------------------- /treeMN-non-hierachical/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN-non-hierachical/main.py -------------------------------------------------------------------------------- /treeMN-non-hierachical/params.py: -------------------------------------------------------------------------------- 1 | use_cuda = True 2 | -------------------------------------------------------------------------------- /treeMN-non-hierachical/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN-non-hierachical/predict.py -------------------------------------------------------------------------------- /treeMN-non-hierachical/treeMN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN-non-hierachical/treeMN.py -------------------------------------------------------------------------------- /treeMN/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN/load_data.py -------------------------------------------------------------------------------- /treeMN/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN/main.py -------------------------------------------------------------------------------- /treeMN/params.py: -------------------------------------------------------------------------------- 1 | use_cuda = True 2 | -------------------------------------------------------------------------------- /treeMN/treeMN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJULearning/TreeAttention/HEAD/treeMN/treeMN.py --------------------------------------------------------------------------------