File: test_gpu_training_continuation.py

package info (click to toggle)
xgboost 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 13,796 kB
  • sloc: cpp: 67,502; python: 35,503; java: 4,676; ansic: 1,426; sh: 1,320; xml: 1,197; makefile: 204; javascript: 19
file content (12 lines) | stat: -rw-r--r-- 377 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
 
import numpy as np
import pytest

from xgboost.testing.continuation import run_training_continuation_model_output

rng = np.random.RandomState(1994)


class TestGPUTrainingContinuation:
    @pytest.mark.parametrize("tree_method", ["hist", "approx"])
    def test_model_output(self, tree_method: str) -> None:
        run_training_continuation_model_output("cuda", tree_method)