Shypd.aiThe largest open index of AI toolsDeepGBM is a deep learning framework distilled by GBDT for online prediction tasks, designed to improve prediction accuracy in real-time applications. It combines the strengths of GBDT and deep learning for enhanced performance.
Trending
Also listed in
About
DeepGBM is a deep learning framework specifically designed for online prediction tasks, leveraging the power of Gradient Boosting Decision Trees (GBDT) for distillation. Presented at KDD'2019, this framework aims to significantly improve prediction accuracy in real-time scenarios. It integrates GBDT-based models, specifically LightGBM, with PyTorch-based neural networks. The project includes comprehensive code for data preprocessing, baseline model implementations, and the proposed DeepGBM model. Users can prepare their data in CSV format, process it through encoders, and then load numerical and categorical data for training. The framework supports training GBDT2NN or the full DeepGBM model, offering flexibility for different prediction needs.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
