66 lines
1.6 KiB
C++
66 lines
1.6 KiB
C++
// Copyright (c) 2015 by Contributors
|
|
// This file contains the customization implementations of R module
|
|
// to change behavior of libxgboost
|
|
|
|
#include <xgboost/logging.h>
|
|
#include "src/common/random.h"
|
|
#include "./xgboost_R.h"
|
|
|
|
// redirect the messages to R's console.
|
|
namespace dmlc {
|
|
void CustomLogMessage::Log(const std::string& msg) {
|
|
Rprintf("%s\n", msg.c_str());
|
|
}
|
|
} // namespace dmlc
|
|
|
|
// implements rabit error handling.
|
|
extern "C" {
|
|
void XGBoostAssert_R(int exp, const char *fmt, ...);
|
|
void XGBoostCheck_R(int exp, const char *fmt, ...);
|
|
}
|
|
|
|
namespace rabit {
|
|
namespace utils {
|
|
extern "C" {
|
|
void (*Printf)(const char *fmt, ...) = Rprintf;
|
|
void (*Assert)(int exp, const char *fmt, ...) = XGBoostAssert_R;
|
|
void (*Check)(int exp, const char *fmt, ...) = XGBoostCheck_R;
|
|
void (*Error)(const char *fmt, ...) = error;
|
|
}
|
|
}
|
|
}
|
|
|
|
namespace xgboost {
|
|
ConsoleLogger::~ConsoleLogger() {
|
|
dmlc::CustomLogMessage::Log(log_stream_.str());
|
|
}
|
|
TrackerLogger::~TrackerLogger() {
|
|
dmlc::CustomLogMessage::Log(log_stream_.str());
|
|
}
|
|
} // namespace xgboost
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
|
|
// redirect the nath functions.
|
|
bool CheckNAN(double v) {
|
|
return ISNAN(v);
|
|
}
|
|
double LogGamma(double v) {
|
|
return lgammafn(v);
|
|
}
|
|
|
|
// customize random engine.
|
|
void CustomGlobalRandomEngine::seed(CustomGlobalRandomEngine::result_type val) {
|
|
// ignore the seed
|
|
}
|
|
|
|
// use R's PRNG to replacd
|
|
CustomGlobalRandomEngine::result_type
|
|
CustomGlobalRandomEngine::operator()() {
|
|
return static_cast<result_type>(
|
|
std::floor(unif_rand() * CustomGlobalRandomEngine::max()));
|
|
}
|
|
} // namespace common
|
|
} // namespace xgboost
|