File: test_gpu_parse_tree.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 (14 lines) | stat: -rw-r--r-- 343 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
 
import sys

sys.path.append("tests/python")
from test_parse_tree import TestTreesToDataFrame


def test_tree_to_df_categorical():
    cputest = TestTreesToDataFrame()
    cputest.run_tree_to_df_categorical("gpu_hist")


def test_split_value_histograms():
    cputest = TestTreesToDataFrame()
    cputest.run_split_value_histograms("gpu_hist")