hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Release] Change all cpus_per_actor in xgboost test. (#18717)

Browse source
This commit is contained in:
xwjiang2010 2021-09-17 12:57:21 -07:00 • committed by GitHub
parent 2c92f737f9
commit 09e760a1fd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
release/golden_notebook_tests/workloads/dask_xgboost_test.py
View file

@ -103,7 +103,7 @@ def main():
bst = train_xgboost(
config, df_train, df_validation, "label",
RayParams(max_actor_restarts=1, cpus_per_actor=8, num_actors=4))
RayParams(max_actor_restarts=1, cpus_per_actor=4, num_actors=4))
tune_xgboost(df_train, df_validation, "label")
inference_df = RayDMatrix(
df_train[sorted(df_train.columns)], ignore=["label", "partition"])