Skip to content

Commit 186a0df

Browse files
author
Siddharth Ancha
committed
Enabled flexibility in stan lang for bdmc - can specify dimensions etc. in data block
1 parent fe7e37d commit 186a0df

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

src/cmdstan/command.hpp

+13-7
Original file line numberDiff line numberDiff line change
@@ -1046,18 +1046,21 @@ namespace stan {
10461046

10471047
std::vector<double> vars_param_prior;
10481048
std::vector<double> vars_param_posterior;
1049-
std::vector<double> vars_data;
1049+
std::vector<double> vars_data_r;
1050+
std::vector<int> vars_data_i;
10501051

10511052
if (load_file == "") {
10521053
stan::bdmc::sample_data_and_params(model,
10531054
vars_param_posterior,
1054-
vars_data,
1055+
vars_data_r,
1056+
vars_data_i,
10551057
base_rng);
10561058
} else {
10571059
stan::bdmc::load_exact_sample (load_file,
10581060
vars_param_prior,
10591061
vars_param_posterior,
1060-
vars_data,
1062+
vars_data_r,
1063+
vars_data_i,
10611064
model);
10621065
}
10631066

@@ -1068,21 +1071,24 @@ namespace stan {
10681071
// prior sample
10691072
Eigen::VectorXd prior_params;
10701073
if (load_file == "") {
1071-
std::vector<double> dummy_data;
1074+
std::vector<double> dummy_data_r;
1075+
std::vector<int> dummy_data_i;
10721076
stan::bdmc::sample_data_and_params(model,
10731077
vars_param_prior,
1074-
dummy_data,
1078+
dummy_data_r,
1079+
dummy_data_i,
10751080
base_rng);
10761081
}
10771082
stan::bdmc::set_params(model, prior_params, vars_param_prior);
10781083

1079-
if (sample_data) stan::bdmc::set_data(model, vars_data);
1084+
if (sample_data) stan::bdmc::set_data(model, vars_data_r, vars_data_i);
10801085

10811086
if (load_file == "" && save_file != "") {
10821087
stan::bdmc::save_exact_sample (save_file,
10831088
vars_param_prior,
10841089
vars_param_posterior,
1085-
vars_data);
1090+
vars_data_r,
1091+
vars_data_i);
10861092
}
10871093

10881094
// Warm up

0 commit comments

Comments
 (0)