Fix nvflare horizontal demo (#9124)

2023-05-05 01:48:22 -07:00
parent 47b3cb6fb7
commit 250b22dd22
3 changed files with 40 additions and 11 deletions
--- a/demo/nvflare/horizontal/custom/trainer.py
+++ b/demo/nvflare/horizontal/custom/trainer.py
@@ -63,8 +63,8 @@ class XGBoostTrainer(Executor):
        }
        with xgb.collective.CommunicatorContext(**communicator_env):
            # Load file, file will not be sharded in federated mode.
-            dtrain = xgb.DMatrix('agaricus.txt.train')
-            dtest = xgb.DMatrix('agaricus.txt.test')
+            dtrain = xgb.DMatrix('agaricus.txt.train?format=libsvm')
+            dtest = xgb.DMatrix('agaricus.txt.test?format=libsvm')

            # Specify parameters via map, definition are same as c++ version
            param = {'max_depth': 2, 'eta': 1, 'objective': 'binary:logistic'}