Merge pull request #5 from zfb132/main

fix bug for GBT
uav4geo · Mar 21, 2023 · 9291331 · 9291331
2 parents 97d016e + 11eb7c7
commit 9291331
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -4,6 +4,7 @@ FROM ubuntu:22.04 as builder
 ENV TZ=Europe/Rome
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
 RUN apt-get update && apt install -y --fix-missing --no-install-recommends \
+    git \
     build-essential \
     software-properties-common \
     cmake \

diff --git a/pcclassify.cpp b/pcclassify.cpp
@@ -59,7 +59,7 @@ int main(int argc, char **argv){
 
         ClassifierType ctype = fingerprint(modelFile);
         #ifndef WITH_GBT
-        if (ctype == GradientBoostedTrees) throw std::runtime_error(modelFile + " is a GBT model but GBT support has not been built (try building with -DWITH_GBT=ON)") << std::endl;
+        if (ctype == GradientBoostedTrees) throw std::runtime_error(modelFile + " is a GBT model but GBT support has not been built (try building with -DWITH_GBT=ON)");
         #endif
 
         std::cout << "Model: " << (ctype == RandomForest ? "Random Forest" : "Gradient Boosted Trees") << std::endl;