Replaced xgb model option gpu_id with 'device' to work with more recent versions of xgboost package.

Author: mcloughlin2

atomsci/ddm/pipeline/model_wrapper.py

@@ -2093,8 +2093,7 @@ def make_dc_model(self, model_dir):
                                          missing=np.nan,
                                          importance_type='gain',
                                          n_jobs=-1,
-                                         gpu_id = -1,
-                                         n_gpus = 0,
+                                         device='cpu',
                                          max_bin = 16,
                                          )
         else:
@@ -2127,9 +2126,8 @@ def make_dc_model(self, model_dir):
                                           random_state=self.seed,
                                           importance_type='gain',
                                           missing=np.nan,
-                                          gpu_id = -1,
+                                          device='cpu',
                                           n_jobs=-1,                                          
-                                          n_gpus = 0,
                                           max_bin = 16,
                                          )
 
@@ -2241,8 +2239,7 @@ def reload_model(self, reload_dir):
                                          missing=np.nan,
                                          importance_type='gain',
                                          n_jobs=-1,
-                                         gpu_id = -1,
-                                         n_gpus = 0,
+                                         device='cpu',
                                          max_bin = 16,
                                          )
         else:
@@ -2265,9 +2262,8 @@ def reload_model(self, reload_dir):
                                          random_state=self.seed, 
                                          importance_type='gain',
                                          missing=np.nan,
-                                         gpu_id = -1,
+                                         device='cpu',
                                          n_jobs=-1,                                          
-                                         n_gpus = 0,
                                          max_bin = 16,
                                          )
 
@@ -2988,4 +2984,4 @@ def get_model_specific_metadata(self):
                     weight_decay_penalty_type=self.params.weight_decay_penalty_type
         )
         model_spec_metadata = dict(nn_specific = nn_metadata)
-        return model_spec_metadata
+        return model_spec_metadata