Skip to content

First implementation of BDTs from scikit-learn...

Steffen Korn requested to merge Adding_BDTs into master

First implementation of BDTs from scikit-learn. The BDT support is added through a BDTModel. The current implementation builds an additive model in a forward stage-wise fashion; it allows for the optimization of arbitrary differentiable loss functions.

In addition DNNPlotHandler is now renamed to ClassificationPlotHandler since both DNNs and BDTs are now handled by this plot handler. The Types for models is now Classification-DNN or Classification-BDT instead of Classifiaction. This requires a change in the config file and the renaming of already produced models (i.e. Classification -> Classification-DNN)

Closes #21 (closed).

Edited by Steffen Korn

Merge request reports

Loading