added scenarii column to lda file (0 is obs. data)

diyabc · Feb 9, 2021 · 40c206f · 40c206f
1 parent f186647
commit 40c206f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/ModelChoice.cpp b/src/ModelChoice.cpp
@@ -74,11 +74,14 @@ ModelChoiceResults ModelChoice_fun(Reftable<MatrixType> &myread,
  const std::string& lda_filename = outfile + ".lda";
  std::ofstream lda_file;
  if (!quiet) {
- MatrixXd ldastatobs = statobs(all,lastN(data_extended.cols()));
- MatrixXd toprint(n+ldastatobs.rows(),data_extended.cols());
+ MatrixXd ldastatobs = statobs(all,lastN(K-1));
+ MatrixXd toprint(n+num_samples,K-1);
  toprint << ldastatobs, data_extended;
+ toprint.conservativeResize(toprint.rows(), toprint.cols()+1);
+ toprint.block(0,K-1,num_samples,1) = MatrixXd::Zero(num_samples,1);
+ for(auto i = 0; i < n; i++) toprint(num_samples+i,K-1) = myread.scenarios[i] + 1; 
  lda_file.open(lda_filename, std::ios::out);
- lda_file << "# First lines (" << ldastatobs.rows() << ") are observed data" << std::endl;
+ lda_file << "# First lines (" << num_samples << ") are observed data" << std::endl;
  lda_file << toprint << std::endl;
  }
  lda_file.close();