Added support for boolean in proc_params

ZiyueXu77 · Jun 4, 2024 · 47deca8 · 47deca8
1 parent 7f3472e
commit 47deca8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/dmlc-core b/dmlc-core
diff --git a/gputreeshap b/gputreeshap
diff --git a/src/collective/communicator.cc b/src/collective/communicator.cc
@@ -35,8 +35,10 @@ std::map<std::string, std::string> json_to_map(xgboost::Json const& config, std:
     } else if (value->Type() == xgboost::Value::ValueKind::kNumber) {
       auto num = reinterpret_cast<xgboost::Number *>(value)->GetNumber();
       text = std::to_string(num);
+    } else if (value->Type() == xgboost::Value::ValueKind::kBoolean) {
+        text = reinterpret_cast<xgboost::Boolean *>(value)->GetBoolean() ? "true" : "false";
     } else {
-      text = "Unsupported type ";
+      text = "Unsupported type";
     }
     params[entry.first] = text;
   }
+1 −6		include/dmlc/config.h
+6 −3		src/data/csv_parser.h
+4 −4		tracker/dmlc_tracker/tracker.py
+1 −7		.gitignore
+1 −9		CMakeLists.txt
+0 −2,576		Doxyfile.in
+12 −45		GPUTreeShap/gpu_treeshap.h
+8 −7		README.md
+10 −24		benchmark/benchmark_gpu_treeshap.cu
+0 −6		build.sh
+25 −5		ci/gpu/build.sh
+0 −33		cmake/doxygen.cmake
+0 −3		example/example.cu
+1 −6		tests/test_gpu_treeshap.cu
+2 −5		tests/test_utils.h