Ensure the dataset includes all necessary features for the Bayesian regression model including test and sample sets