Permalink
Browse files

Dumping Sigma in Bonsai for SeeDot

  • Loading branch information...
Aditya Kusupati
Aditya Kusupati committed Dec 28, 2018
1 parent 80d9135 commit b943a6d86aa16d2d96f67a9f029185e6432445c4
Showing with 5 additions and 0 deletions.
  1. +5 −0 cpp/src/Bonsai/BonsaiTrainer.cpp
@@ -645,6 +645,11 @@ void BonsaiTrainer::dumpModelMeanStd(const std::string& currResultsPath)

writeMatrixInASCII(mean, params_path, "Mean");
writeMatrixInASCII(stdDev, params_path, "Std");

std::ofstream sigmaDumper(params_path + "/Sigma", std::ofstream::out);
sigmaDumper << model.hyperParams.Sigma;
sigmaDumper.close();

LOG_INFO(currResultsPath);
}

0 comments on commit b943a6d

Please sign in to comment.