Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add callback interface to re-direct console output (#3438)

Browse Source 
* Add callback interface to re-direct console output

* Exempt TrackerLogger from custom logging

* Fix lint
This commit is contained in:
Philip Hyunsu Cho
2018-07-05 11:32:30 -07:00
committed by GitHub
parent 45bf4fbffb
commit 48d6e68690
7 changed files with 64 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/logging.cc
View File

@@ -8,16 +8,24 @@
#include <iostream>
#include "./common/sync.h"
namespace xgboost {
#if !defined(XGBOOST_STRICT_R_MODE) || XGBOOST_STRICT_R_MODE == 0
// Override logging mechanism for non-R interfaces
void dmlc::CustomLogMessage::Log(const std::string& msg) {
const xgboost::LogCallbackRegistry* registry
= xgboost::LogCallbackRegistryStore::Get();
auto callback = registry->Get();
callback(msg.c_str());
}
#if XGBOOST_CUSTOMIZE_LOGGER == 0
namespace xgboost {
ConsoleLogger::~ConsoleLogger() {
std::cerr << log_stream_.str() << std::endl;
dmlc::CustomLogMessage::Log(log_stream_.str());
}
TrackerLogger::~TrackerLogger() {
log_stream_ << '\n';
rabit::TrackerPrint(log_stream_.str());
}
#endif
} // namespace xgboost
#endif