Vicentini Filippo · 579766bf · 579766bf · 579766bf · 579766bf · 579766bf
--- a/NetKet/Unsupervised/quantum_state_reconstruction.hpp
+++ b/NetKet/Unsupervised/quantum_state_reconstruction.hpp
+ }
+
+ void Run() {
+   std::vector<Eigen::VectorXd> batchSamples;
+   std::vector<int> batchBases;
+   opt_.Reset();
+
+   InitSweeps();
+   std::uniform_int_distribution<int> distribution(
+       0, trainingSamples_.size() - 1);
+
+   for (int i = 0; i < niter_opt_; i++) {
+     int index;
+     batchSamples.resize(batchsize_node_);
+     batchBases.resize(batchsize_node_);
+
--- a/NetKet/Unsupervised/quantum_state_reconstruction.hpp
+++ b/NetKet/Unsupervised/quantum_state_reconstruction.hpp
+     }
+     Gradient(batchSamples, batchBases);
+     UpdateParameters();
+     PrintOutput(i);
+   }
+ }
+
+ void UpdateParameters() {
+   auto pars = psi_.GetParameters();
+   opt_.Update(grad_, pars);
+   SendToAll(pars);
+   psi_.SetParameters(pars);
+   MPI_Barrier(MPI_COMM_WORLD);
+ }
+
+ void RotateGradient(int b_index, const Eigen::VectorXd &state,