From d501fd299cebe0346d2c984b4590fbbf451156ba Mon Sep 17 00:00:00 2001
From: vchekali <vchekali@cern.ch>
Date: Tue, 29 May 2018 12:32:00 +0200
Subject: [PATCH] big files to param files
---
Calo/CaloTools/src/GammaPi0XGBoostTool.cpp | 20 +++++++++++++-------
1 file changed, 13 insertions(+), 7 deletions(-)
diff --git a/Calo/CaloTools/src/GammaPi0XGBoostTool.cpp b/Calo/CaloTools/src/GammaPi0XGBoostTool.cpp
index 758c96d3912..df69a202ff9 100644
--- a/Calo/CaloTools/src/GammaPi0XGBoostTool.cpp
+++ b/Calo/CaloTools/src/GammaPi0XGBoostTool.cpp
@@ -14,6 +14,7 @@
// local
#include "GammaPi0XGBoostTool.h"
+#include <fstream>
using namespace LHCb;
using namespace Gaudi::Units;
@@ -148,13 +149,18 @@ StatusCode GammaPi0XGBoostTool::initialize() {
// TMVA discriminant
std::string cur_path(getenv("PWD"));
+
+ // Get path to a root directory with models
+ const std::string paramEnv = "PARAMFILESROOT";
+ std::string paramRoot = std::string(getenv(paramEnv.c_str()));
+ // Load Iron Transformer string filename = paramRoot + "/data/MC15TuneCatBoostV1/iron_transform.txt";
- m_xgb0 = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/0_simpl.model");
- m_xgb1 = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/1_simpl.model");
- m_xgb2 = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/2_simpl.model");
- m_xgb0_b = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/0_bound.model");
- m_xgb1_b = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/1_bound.model");
- m_xgb2_b = std::make_unique<XGBClassifierPhPi0>(cur_path + "/models/classes/2_bound.model");
+ m_xgb0 = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/0_simpl.model");
+ m_xgb1 = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/1_simpl.model");
+ m_xgb2 = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/2_simpl.model");
+ m_xgb0_b = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/0_bound.model");
+ m_xgb1_b = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/1_bound.model");
+ m_xgb2_b = std::make_unique<XGBClassifierPhPi0>(paramRoot + "/2_bound.model");
return StatusCode::SUCCESS;
}
@@ -300,4 +306,4 @@ std::vector<std::vector<double>> GammaPi0XGBoostTool::GetCluster(const LHCb::Cal
}
}
return vector_cells;
-}
\ No newline at end of file
+}
--
GitLab