xgboost/tests/python-gpu/test_gpu_parse_tree.py
Jiaming Yuan d6ea5cc1ed
Cover approx tree method for categorical data tests. (#7569)
* Add tree to df tests.
* Add plotting tests.
* Add histogram tests.
2022-01-16 11:31:40 +08:00

15 lines
343 B
Python

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")