Skip to content
/ ABDF Public

Attention-based Deep Forest. Supplementary code for paper "Flexible deep forest classifier with multi-head attention"

Notifications You must be signed in to change notification settings

NTAILab/ABDF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Attention-based Deep Forest (ABDF)

This repository contains the code base for the Attention-based Deep Forest model, proposed in the article.

The code is written in Python 3 language with usage of the Bosk framework, so the model is compatible with the scikit-learn API. All the necessary code is located in the only script. To run it, install the Bosk and Pytorch to your Python environment.

About

Attention-based Deep Forest. Supplementary code for paper "Flexible deep forest classifier with multi-head attention"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages