Skip to content

GitLab

Unverified Commit 0f3d90e7 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

increase the threshold of using predict_by_map

parent ed3d774c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/application/predictor.hpp
View file @ 0f3d90e7
...@@ -60,8 +60,8 @@ public: ...@@ -60,8 +60,8 @@ public:
num_pred_one_row_ = boosting_->NumPredictOneRow(num_iteration, is_predict_leaf_index, is_predict_contrib); num_pred_one_row_ = boosting_->NumPredictOneRow(num_iteration, is_predict_leaf_index, is_predict_contrib);
num_feature_ = boosting_->MaxFeatureIdx() + 1; num_feature_ = boosting_->MaxFeatureIdx() + 1;
predict_buf_ = std::vector<std::vector<double>>(num_threads_, std::vector<double>(num_feature_, 0.0f)); predict_buf_ = std::vector<std::vector<double>>(num_threads_, std::vector<double>(num_feature_, 0.0f));
const int kFeatureThreshold = 20000; const int kFeatureThreshold = 100000;
const size_t KSparseThreshold = static_cast<size_t>(0.02 * num_feature_); const size_t KSparseThreshold = static_cast<size_t>(0.01 * num_feature_);
if (is_predict_leaf_index) { if (is_predict_leaf_index) {
predict_fun_ = [this, kFeatureThreshold, KSparseThreshold](const std::vector<std::pair<int, double>>& features, double* output) { predict_fun_ = [this, kFeatureThreshold, KSparseThreshold](const std::vector<std::pair<int, double>>& features, double* output) {
int tid = omp_get_thread_num(); int tid = omp_get_thread_num();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment