boosting.cpp 312 Bytes
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#include <LightGBM/boosting.h>
#include "gbdt.h"

namespace LightGBM {

Boosting* Boosting::CreateBoosting(BoostingType type,
                         const BoostingConfig* config) {
  if (type == BoostingType::kGBDT) {
    return new GBDT(config);
  } else {
    return nullptr;
  }
}

}  // namespace LightGBM