/dports/misc/py-xgboost/xgboost-1.5.1/src/linear/ |
H A D | updater_coordinate.cc | 48 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 55 in_gpair->ConstHostVector(), p_fmat); in Update() 60 dbias, &in_gpair->HostVector(), p_fmat); in Update() 63 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 70 (i, *model, group_idx, in_gpair->ConstHostVector(), p_fmat, in Update() 73 this->UpdateFeature(fidx, group_idx, &in_gpair->HostVector(), p_fmat, model); in Update() 80 DMatrix *p_fmat, gbm::GBLinearModel *model) { in UpdateFeature() argument 84 GetGradientParallel(group_idx, ngroup, fidx, *in_gpair, p_fmat); in UpdateFeature() 90 UpdateResidualParallel(fidx, group_idx, ngroup, dw, in_gpair, p_fmat); in UpdateFeature()
|
H A D | coordinate_common.h | 83 DMatrix *p_fmat) { in GetGradient() argument 85 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in GetGradient() 113 DMatrix *p_fmat) { in GetGradientParallel() argument 115 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in GetGradientParallel() 149 const auto ndata = static_cast<bst_omp_uint>(p_fmat->Info().num_row_); in GetBiasGradientParallel() 177 DMatrix *p_fmat) { in UpdateResidualParallel() argument 179 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in UpdateResidualParallel() 208 DMatrix *p_fmat) { in UpdateBiasResidualParallel() argument 210 const auto ndata = static_cast<bst_omp_uint>(p_fmat->Info().num_row_); in UpdateBiasResidualParallel() 356 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in NextFeature() [all …]
|
H A D | updater_shotgun.cc | 35 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 44 in_gpair->ConstHostVector(), p_fmat); in Update() 48 UpdateBiasResidualParallel(gid, ngroup, dbias, &in_gpair->HostVector(), p_fmat); in Update() 52 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 54 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in Update() 62 (i, *model, 0, in_gpair->ConstHostVector(), p_fmat, param_.reg_alpha_denorm, in Update()
|
H A D | updater_gpu_coordinate.cu | 50 void LazyInitDevice(DMatrix *p_fmat, const LearnerModelParam &model_param) { in LazyInitDevice() argument 53 num_row_ = static_cast<size_t>(p_fmat->Info().num_row_); in LazyInitDevice() 55 CHECK(p_fmat->SingleColBlock()); in LazyInitDevice() 56 SparsePage const& batch = *(p_fmat->GetBatches<CSCPage>().begin()); in LazyInitDevice() 96 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 100 this->LazyInitDevice(p_fmat, *(model->learner_model_param)); in Update() 112 this->UpdateBias(p_fmat, model); in Update() 115 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 123 i, *model, group_idx, in_gpair->ConstHostVector(), p_fmat, in Update() 132 void UpdateBias(DMatrix *p_fmat, gbm::GBLinearModel *model) { in UpdateBias() argument
|
/dports/misc/xgboost/xgboost-1.5.1/src/linear/ |
H A D | updater_coordinate.cc | 48 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 55 in_gpair->ConstHostVector(), p_fmat); in Update() 60 dbias, &in_gpair->HostVector(), p_fmat); in Update() 63 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 70 (i, *model, group_idx, in_gpair->ConstHostVector(), p_fmat, in Update() 73 this->UpdateFeature(fidx, group_idx, &in_gpair->HostVector(), p_fmat, model); in Update() 80 DMatrix *p_fmat, gbm::GBLinearModel *model) { in UpdateFeature() argument 84 GetGradientParallel(group_idx, ngroup, fidx, *in_gpair, p_fmat); in UpdateFeature() 90 UpdateResidualParallel(fidx, group_idx, ngroup, dw, in_gpair, p_fmat); in UpdateFeature()
|
H A D | coordinate_common.h | 83 DMatrix *p_fmat) { in GetGradient() argument 85 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in GetGradient() 113 DMatrix *p_fmat) { in GetGradientParallel() argument 115 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in GetGradientParallel() 149 const auto ndata = static_cast<bst_omp_uint>(p_fmat->Info().num_row_); in GetBiasGradientParallel() 177 DMatrix *p_fmat) { in UpdateResidualParallel() argument 179 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in UpdateResidualParallel() 208 DMatrix *p_fmat) { in UpdateBiasResidualParallel() argument 210 const auto ndata = static_cast<bst_omp_uint>(p_fmat->Info().num_row_); in UpdateBiasResidualParallel() 356 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in NextFeature() [all …]
|
H A D | updater_shotgun.cc | 35 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 44 in_gpair->ConstHostVector(), p_fmat); in Update() 48 UpdateBiasResidualParallel(gid, ngroup, dbias, &in_gpair->HostVector(), p_fmat); in Update() 52 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 54 for (const auto &batch : p_fmat->GetBatches<CSCPage>()) { in Update() 62 (i, *model, 0, in_gpair->ConstHostVector(), p_fmat, param_.reg_alpha_denorm, in Update()
|
H A D | updater_gpu_coordinate.cu | 50 void LazyInitDevice(DMatrix *p_fmat, const LearnerModelParam &model_param) { in LazyInitDevice() argument 53 num_row_ = static_cast<size_t>(p_fmat->Info().num_row_); in LazyInitDevice() 55 CHECK(p_fmat->SingleColBlock()); in LazyInitDevice() 56 SparsePage const& batch = *(p_fmat->GetBatches<CSCPage>().begin()); in LazyInitDevice() 96 void Update(HostDeviceVector<GradientPair> *in_gpair, DMatrix *p_fmat, in Update() argument 100 this->LazyInitDevice(p_fmat, *(model->learner_model_param)); in Update() 112 this->UpdateBias(p_fmat, model); in Update() 115 selector_->Setup(*model, in_gpair->ConstHostVector(), p_fmat, in Update() 123 i, *model, group_idx, in_gpair->ConstHostVector(), p_fmat, in Update() 132 void UpdateBias(DMatrix *p_fmat, gbm::GBLinearModel *model) { in UpdateBias() argument
|
/dports/misc/py-xgboost/xgboost-1.5.1/src/gbm/ |
H A D | gblinear.cc | 137 void DoBoost(DMatrix *p_fmat, in DoBoost() argument 143 this->LazySumWeights(p_fmat); in DoBoost() 176 void PredictContribution(DMatrix* p_fmat, in PredictContribution() argument 190 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictContribution() 214 void PredictInteractionContributions(DMatrix* p_fmat, in PredictInteractionContributions() argument 223 contribs.resize(p_fmat->Info().num_row_ * nelements * in PredictInteractionContributions() 265 void PredictBatchInternal(DMatrix *p_fmat, in PredictBatchInternal() argument 273 preds.resize(p_fmat->Info().num_row_ * ngroup); in PredictBatchInternal() 274 for (const auto &page : p_fmat->GetBatches<SparsePage>()) { in PredictBatchInternal() 315 void LazySumWeights(DMatrix *p_fmat) { in LazySumWeights() argument [all …]
|
H A D | gbtree.cc | 219 void GBTree::DoBoost(DMatrix* p_fmat, in DoBoost() argument 224 ConfigureWithKnownData(this->cfg_, p_fmat); in DoBoost() 238 BoostNewTrees(in_gpair, p_fmat, 0, &ret); in DoBoost() 257 BoostNewTrees(&tmp, p_fmat, gid, &ret); in DoBoost() 316 DMatrix *p_fmat, in BoostNewTrees() argument 352 CHECK_EQ(gpair->Size(), p_fmat->Info().num_row_) in BoostNewTrees() 356 up->Update(gpair, p_fmat, new_trees); in BoostNewTrees() 477 void GBTree::PredictBatch(DMatrix* p_fmat, in PredictBatch() argument 747 size_t n_rows = p_fmat->Info().num_row_; in PredictBatchImpl() 765 void PredictBatch(DMatrix* p_fmat, in PredictBatch() argument [all …]
|
H A D | gbtree.h | 217 void DoBoost(DMatrix* p_fmat, 266 void PredictBatch(DMatrix *p_fmat, PredictionCacheEntry *out_preds, 379 void PredictLeaf(DMatrix* p_fmat, in PredictLeaf() argument 386 this->GetPredictor()->PredictLeaf(p_fmat, out_preds, model_, tree_end); in PredictLeaf() 389 void PredictContribution(DMatrix* p_fmat, in PredictContribution() argument 400 p_fmat, out_contribs, model_, tree_end, nullptr, approximate); in PredictContribution() 404 DMatrix *p_fmat, HostDeviceVector<bst_float> *out_contribs, in PredictInteractionContributions() argument 413 p_fmat, out_contribs, model_, tree_end, nullptr, approximate); in PredictInteractionContributions() 428 DMatrix *p_fmat,
|
/dports/misc/xgboost/xgboost-1.5.1/src/gbm/ |
H A D | gblinear.cc | 137 void DoBoost(DMatrix *p_fmat, in DoBoost() argument 143 this->LazySumWeights(p_fmat); in DoBoost() 176 void PredictContribution(DMatrix* p_fmat, in PredictContribution() argument 190 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictContribution() 214 void PredictInteractionContributions(DMatrix* p_fmat, in PredictInteractionContributions() argument 223 contribs.resize(p_fmat->Info().num_row_ * nelements * in PredictInteractionContributions() 265 void PredictBatchInternal(DMatrix *p_fmat, in PredictBatchInternal() argument 273 preds.resize(p_fmat->Info().num_row_ * ngroup); in PredictBatchInternal() 274 for (const auto &page : p_fmat->GetBatches<SparsePage>()) { in PredictBatchInternal() 315 void LazySumWeights(DMatrix *p_fmat) { in LazySumWeights() argument [all …]
|
H A D | gbtree.cc | 219 void GBTree::DoBoost(DMatrix* p_fmat, in DoBoost() argument 224 ConfigureWithKnownData(this->cfg_, p_fmat); in DoBoost() 238 BoostNewTrees(in_gpair, p_fmat, 0, &ret); in DoBoost() 257 BoostNewTrees(&tmp, p_fmat, gid, &ret); in DoBoost() 316 DMatrix *p_fmat, in BoostNewTrees() argument 352 CHECK_EQ(gpair->Size(), p_fmat->Info().num_row_) in BoostNewTrees() 356 up->Update(gpair, p_fmat, new_trees); in BoostNewTrees() 477 void GBTree::PredictBatch(DMatrix* p_fmat, in PredictBatch() argument 747 size_t n_rows = p_fmat->Info().num_row_; in PredictBatchImpl() 765 void PredictBatch(DMatrix* p_fmat, in PredictBatch() argument [all …]
|
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/linear/ |
H A D | test_linear.cc | 18 auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix(); in TEST() local 31 p_fmat->Info().num_row_, xgboost::GradientPair(-5, 1.0)); in TEST() 34 updater->Update(&gpair, p_fmat.get(), &model, gpair.Size()); in TEST() 54 auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix(); in TEST() local 66 p_fmat->Info().num_row_, xgboost::GradientPair(-5, 1.0)); in TEST() 69 updater->Update(&gpair, p_fmat.get(), &model, gpair.Size()); in TEST()
|
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/linear/ |
H A D | test_linear.cc | 18 auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix(); in TEST() local 31 p_fmat->Info().num_row_, xgboost::GradientPair(-5, 1.0)); in TEST() 34 updater->Update(&gpair, p_fmat.get(), &model, gpair.Size()); in TEST() 54 auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix(); in TEST() local 66 p_fmat->Info().num_row_, xgboost::GradientPair(-5, 1.0)); in TEST() 69 updater->Update(&gpair, p_fmat.get(), &model, gpair.Size()); in TEST()
|
/dports/misc/py-xgboost/xgboost-1.5.1/src/tree/ |
H A D | updater_histmaker.cc | 28 DMatrix *p_fmat, in Update() argument 109 DMatrix *p_fmat, in UpdateTree() argument 112 this->InitData(gpair, *p_fmat, *p_tree); in UpdateTree() 315 void InitWorkSet(DMatrix *p_fmat, in InitWorkSet() argument 318 if (p_fmat != cache_dmatrix_) { in InitWorkSet() 320 cache_dmatrix_ = p_fmat; in InitWorkSet() 327 DMatrix *p_fmat, in CreateHist() argument 330 const MetaInfo &info = p_fmat->Info(); in CreateHist() 382 DMatrix *p_fmat, in ResetPosAndPropose() argument 385 const MetaInfo &info = p_fmat->Info(); in ResetPosAndPropose() [all …]
|
H A D | updater_colmaker.cc | 167 DMatrix* p_fmat, in Update() argument 170 this->InitData(gpair, *p_fmat); in Update() 171 this->InitNewNode(qexpand_, gpair, *p_fmat, *p_tree); in Update() 173 this->FindSplit(depth, qexpand_, gpair, p_fmat, p_tree); in Update() 174 this->ResetPosition(qexpand_, p_fmat, *p_tree); in Update() 176 this->InitNewNode(newnodes, gpair, *p_fmat, *p_tree); in Update() 488 DMatrix *p_fmat, in FindSplit() argument 493 for (const auto &batch : p_fmat->GetBatches<SortedCSCPage>()) { in FindSplit() 522 DMatrix* p_fmat, in ResetPosition() argument 525 this->SetNonDefaultPosition(qexpand, p_fmat, tree); in ResetPosition() [all …]
|
/dports/misc/xgboost/xgboost-1.5.1/src/tree/ |
H A D | updater_histmaker.cc | 28 DMatrix *p_fmat, in Update() argument 109 DMatrix *p_fmat, in UpdateTree() argument 112 this->InitData(gpair, *p_fmat, *p_tree); in UpdateTree() 315 void InitWorkSet(DMatrix *p_fmat, in InitWorkSet() argument 318 if (p_fmat != cache_dmatrix_) { in InitWorkSet() 320 cache_dmatrix_ = p_fmat; in InitWorkSet() 327 DMatrix *p_fmat, in CreateHist() argument 330 const MetaInfo &info = p_fmat->Info(); in CreateHist() 382 DMatrix *p_fmat, in ResetPosAndPropose() argument 385 const MetaInfo &info = p_fmat->Info(); in ResetPosAndPropose() [all …]
|
H A D | updater_colmaker.cc | 167 DMatrix* p_fmat, in Update() argument 170 this->InitData(gpair, *p_fmat); in Update() 171 this->InitNewNode(qexpand_, gpair, *p_fmat, *p_tree); in Update() 173 this->FindSplit(depth, qexpand_, gpair, p_fmat, p_tree); in Update() 174 this->ResetPosition(qexpand_, p_fmat, *p_tree); in Update() 176 this->InitNewNode(newnodes, gpair, *p_fmat, *p_tree); in Update() 488 DMatrix *p_fmat, in FindSplit() argument 493 for (const auto &batch : p_fmat->GetBatches<SortedCSCPage>()) { in FindSplit() 522 DMatrix* p_fmat, in ResetPosition() argument 525 this->SetNonDefaultPosition(qexpand, p_fmat, tree); in ResetPosition() [all …]
|
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/tree/ |
H A D | test_quantile_hist.cc | 36 DMatrix* p_fmat, in TestInitData() 38 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitData() 48 const size_t num_row = p_fmat->Info().num_row_; in TestInitData() 49 const size_t num_col = p_fmat->Info().num_col_; in TestInitData() 70 for (const auto& batch : p_fmat->GetBatches<xgboost::SparsePage>()) { in TestInitData() 101 DMatrix* p_fmat, in TestInitDataSampling() 119 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitDataSampling() 122 ASSERT_EQ(row_indices_initial.size(), p_fmat->Info().num_row_); in TestInitDataSampling() 130 p_fmat->Info().num_row_); in TestInitDataSampling() 137 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitDataSampling() [all …]
|
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/tree/ |
H A D | test_quantile_hist.cc | 36 DMatrix* p_fmat, in TestInitData() 38 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitData() 48 const size_t num_row = p_fmat->Info().num_row_; in TestInitData() 49 const size_t num_col = p_fmat->Info().num_col_; in TestInitData() 70 for (const auto& batch : p_fmat->GetBatches<xgboost::SparsePage>()) { in TestInitData() 101 DMatrix* p_fmat, in TestInitDataSampling() 119 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitDataSampling() 122 ASSERT_EQ(row_indices_initial.size(), p_fmat->Info().num_row_); in TestInitDataSampling() 130 p_fmat->Info().num_row_); in TestInitDataSampling() 137 RealImpl::InitData(gmat, *p_fmat, tree, gpair); in TestInitDataSampling() [all …]
|
/dports/misc/py-xgboost/xgboost-1.5.1/src/data/ |
H A D | gradient_index.cc | 128 void GHistIndexMatrix::Init(DMatrix* p_fmat, int max_bins, common::Span<float> hess) { in Init() argument 129 cut = common::SketchOnDMatrix(p_fmat, max_bins, hess); in Init() 137 this->p_fmat = p_fmat; in Init() 139 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in Init() 148 const bool isDense = p_fmat->IsDense(); in Init() 151 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in Init()
|
/dports/misc/xgboost/xgboost-1.5.1/src/data/ |
H A D | gradient_index.cc | 128 void GHistIndexMatrix::Init(DMatrix* p_fmat, int max_bins, common::Span<float> hess) { in Init() argument 129 cut = common::SketchOnDMatrix(p_fmat, max_bins, hess); in Init() 137 this->p_fmat = p_fmat; in Init() 139 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in Init() 148 const bool isDense = p_fmat->IsDense(); in Init() 151 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in Init()
|
/dports/misc/py-xgboost/xgboost-1.5.1/src/predictor/ |
H A D | cpu_predictor.cc | 257 size_t total = std::max(p_fmat->Info().num_row_ * p_fmat->Info().num_col_, in PredictDMatrix() 259 double density = static_cast<double>(p_fmat->Info().num_nonzero_) / in PredictDMatrix() 266 for (auto const &batch : p_fmat->GetBatches<SparsePage>()) { in PredictDMatrix() 268 p_fmat->Info().num_row_ * in PredictDMatrix() 409 const MetaInfo& info = p_fmat->Info(); in PredictLeaf() 417 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictLeaf() 440 void PredictContribution(DMatrix *p_fmat, in PredictContribution() argument 450 const MetaInfo& info = p_fmat->Info(); in PredictContribution() 472 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictContribution() 521 DMatrix *p_fmat, HostDeviceVector<bst_float> *out_contribs, in PredictInteractionContributions() argument [all …]
|
/dports/misc/xgboost/xgboost-1.5.1/src/predictor/ |
H A D | cpu_predictor.cc | 257 size_t total = std::max(p_fmat->Info().num_row_ * p_fmat->Info().num_col_, in PredictDMatrix() 259 double density = static_cast<double>(p_fmat->Info().num_nonzero_) / in PredictDMatrix() 266 for (auto const &batch : p_fmat->GetBatches<SparsePage>()) { in PredictDMatrix() 268 p_fmat->Info().num_row_ * in PredictDMatrix() 409 const MetaInfo& info = p_fmat->Info(); in PredictLeaf() 417 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictLeaf() 440 void PredictContribution(DMatrix *p_fmat, in PredictContribution() argument 450 const MetaInfo& info = p_fmat->Info(); in PredictContribution() 472 for (const auto &batch : p_fmat->GetBatches<SparsePage>()) { in PredictContribution() 521 DMatrix *p_fmat, HostDeviceVector<bst_float> *out_contribs, in PredictInteractionContributions() argument [all …]
|