xgboost

Hendrik/xgboost

Fork 0

Commit Graph

Author	SHA1	Message	Date
Rong Ou	c5b229632d	[BREAKING] prevent multi-gpu usage (#4749 ) * prevent multi-gpu usage * fix distributed test * combine gpu predictor tests * set upper bound on n_gpus	2019-08-13 09:11:35 +12:00
Jiaming Yuan	b48f895027	Fix prediction from loaded pickle. (#4516 )	2019-05-30 15:05:09 +08:00
Jiaming Yuan	c589eff941	De-duplicate GPU parameters. (#4454 ) * Only define `gpu_id` and `n_gpus` in `LearnerTrainParam` * Pass LearnerTrainParam through XGBoost vid factory method. * Disable all GPU usage when GPU related parameters are not specified (fixes XGBoost choosing GPU over aggressively). * Test learner train param io. * Fix gpu pickling.	2019-05-29 11:55:57 +08:00

Author

SHA1

Message

Date

Rong Ou

c5b229632d

[BREAKING] prevent multi-gpu usage (#4749 )

* prevent multi-gpu usage

* fix distributed test

* combine gpu predictor tests

* set upper bound on n_gpus

2019-08-13 09:11:35 +12:00

Jiaming Yuan

b48f895027

Fix prediction from loaded pickle. (#4516 )

2019-05-30 15:05:09 +08:00

Jiaming Yuan

c589eff941

De-duplicate GPU parameters. (#4454 )

* Only define `gpu_id` and `n_gpus` in `LearnerTrainParam`
* Pass LearnerTrainParam through XGBoost vid factory method.
* Disable all GPU usage when GPU related parameters are not specified (fixes XGBoost choosing GPU over aggressively).
* Test learner train param io.
* Fix gpu pickling.

2019-05-29 11:55:57 +08:00

3 Commits