
    =y!d                         d dl mZ d Zd ZdS )    )_backendc                 N    t           j        j                            | ||          S )N)r   get_treeclassificationget_tree_state)modeliTree	n_classess      :lib/python3.11/site-packages/onedal/primitives/get_tree.pyget_tree_state_clsr      s'    +::ui! ! !    c                 N    t           j        j                            | |d          S )N   )r   r   
regressionr   )r   r	   s     r   get_tree_state_regr      s     '66ueQGGGr   N)onedalr   r   r    r   r   <module>r      sF   "      ! ! !
H H H H Hr   