add 'null' to missing value (#1064)

add 'null' to missing value

add 'null' to missing value (#1064)
add 'null' to missing value
9ba0cc78 · zhangjin · Guolin Ke · e5eb8560 · 9ba0cc78
Commit 9ba0cc78 authored Nov 16, 2017 by zhangjin Committed by Guolin Ke Nov 16, 2017
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

include/LightGBM/utils/common.h include/LightGBM/utils/common.h +2 -1

No files found.
--- a/include/LightGBM/utils/common.h
+++ b/include/LightGBM/utils/common.h
@@ -248,7 +248,8 @@ inline static const char* Atof(const char* p, double* out) {
    if (cnt > 0) {
      std::string tmp_str(p, cnt);
      std::transform(tmp_str.begin(), tmp_str.end(), tmp_str.begin(), Common::tolower);
-      if (tmp_str == std::string("na") || tmp_str == std::string("nan")) {
+      if (tmp_str == std::string("na") || tmp_str == std::string("nan") ||
+          tmp_str == std::string("null")) {
        *out = NAN;
      } else if (tmp_str == std::string("inf") || tmp_str == std::string("infinity")) {
        *out = sign * 1e308;