The repository contains the code used to produce the results and the figure illustrated in the article "Loss-based prior for tree topologies in BART model"s
The code folder contains the functions for the MCMC algorithm, analyse posterior results, visualise trees.
The file figures_for_paper.R produces Figures 1 to 5
The file Synthetic_example.R produces Figures 6 to 8, Figures 13 to 15, and the results in Table 1
The file Breast_cancer.R produces Figures 9 to 12, and Figures 16 to 19.