Fix parameter loading with training continuation. (#7121)

* Add a demo for training continuation.
This commit is contained in:
Jiaming Yuan
2021-07-23 10:51:47 +08:00
committed by GitHub
parent 41e882f80b
commit 778135f657
5 changed files with 151 additions and 13 deletions

View File

@@ -127,6 +127,12 @@ def test_callbacks_demo():
subprocess.check_call(cmd)
def test_continuation_demo():
script = os.path.join(PYTHON_DEMO_DIR, 'continuation.py')
cmd = ['python', script]
subprocess.check_call(cmd)
# gpu_acceleration is not tested due to covertype dataset is being too huge.
# gamma regression is not tested as it requires running a R script first.
# aft viz is not tested due to ploting is not controled