| 12 | namespace LightGBM { |
| 13 | |
| 14 | TreeLearner* TreeLearner::CreateTreeLearner(const std::string& learner_type, const std::string& device_type, const Config* config) { |
| 15 | if (device_type == std::string("cpu")) { |
| 16 | if (learner_type == std::string("serial")) { |
| 17 | return new SerialTreeLearner(config); |
| 18 | } else if (learner_type == std::string("serial2")) { |
| 19 | return new SerialTreeLearner2(config); |
| 20 | } else if (learner_type == std::string("feature")) { |
| 21 | return new FeatureParallelTreeLearner<SerialTreeLearner>(config); |
| 22 | } else if (learner_type == std::string("data")) { |
| 23 | return new DataParallelTreeLearner<SerialTreeLearner>(config); |
| 24 | } else if (learner_type == std::string("voting")) { |
| 25 | return new VotingParallelTreeLearner<SerialTreeLearner>(config); |
| 26 | } |
| 27 | } else if (device_type == std::string("gpu")) { |
| 28 | if (learner_type == std::string("serial")) { |
| 29 | return new GPUTreeLearner(config); |
| 30 | } else if (learner_type == std::string("feature")) { |
| 31 | return new FeatureParallelTreeLearner<GPUTreeLearner>(config); |
| 32 | } else if (learner_type == std::string("data")) { |
| 33 | return new DataParallelTreeLearner<GPUTreeLearner>(config); |
| 34 | } else if (learner_type == std::string("voting")) { |
| 35 | return new VotingParallelTreeLearner<GPUTreeLearner>(config); |
| 36 | } |
| 37 | } |
| 38 | return nullptr; |
| 39 | } |
| 40 | |
| 41 | void TreeLearner::Train_serial2(Tree* tree, const score_t* gradients, const score_t* hessians) { |
| 42 | // return 0; |
nothing calls this directly
no outgoing calls
no test coverage detected