Skip to content

Commit bdb081a

Browse files
velochyaloctavodia
andauthored
Fix the reference to flexBART
Co-authored-by: Osvaldo A Martin <aloctavodia@gmail.com>
1 parent 11ce6b0 commit bdb081a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_bart/split_rules.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def divide(available_splitting_values, split_value):
7878
class SubsetSplitRule(SplitRule):
7979
"""
8080
Choose a random subset of the categorical values and branch on if the value is within the chosen set.
81-
This is the approach taken in flexBART paper.
81+
This is the approach taken by Sameer K. Deshpande. flexBART: Flexible Bayesian regression trees with categorical predictors. arXiv, `link <https://arxiv.org/abs/2211.04459>`__
8282
"""
8383

8484
@staticmethod

0 commit comments

Comments
 (0)