Optimize DMatrix build time. (#5877)

Co-authored-by: SHVETS, KIRILL <kirill.shvets@intel.com>
2020-08-19 20:37:03 +03:00
parent 29b7fea572
commit 24f2e6c97e
3 changed files with 70 additions and 40 deletions
--- a/python-package/xgboost/core.py
+++ b/python-package/xgboost/core.py
@@ -422,7 +422,7 @@ class DMatrix:                  # pylint: disable=too-many-instance-attributes
            raise TypeError('Input data can not be a list.')

        self.missing = missing if missing is not None else np.nan
-        self.nthread = nthread if nthread is not None else 1
+        self.nthread = nthread if nthread is not None else -1
        self.silent = silent

        # force into void_p, mac need to pass things in as void_p