Ajout de update dans la fonction draw

1 files changed, 4 insertions, 3 deletions

diff --git a/stratified_sampling.hpp b/stratified_sampling.hpp
index e7b3978..81080fc 100644
--- a/stratified_sampling.hpp
+++ b/stratified_sampling.hpp
@@ -49,14 +49,14 @@ template <typename L>
 struct stratified_sampling {
     stratified_sampling(vector<double> p, vector<L> X)
         :p(p), X(X), mean(p.size(), 0), sigma2(p.size(), 0), I(p.size()){};
-    void update(int N);
-    void draw();
+    void draw(int N);
     vector<double> get_mean() const;
     vector<double> get_var() const;
     void print_mean() const;
     void print_sigma() const;
     pair<double,double> estimator() const;
 private:
+		void update(int N);
     vector<double> p;
     vector<L> X;
     vector<int> M;
@@ -111,7 +111,8 @@ void stratified_sampling<L>::update(int Nk) {
 }
 
 template <typename L>
-void stratified_sampling<L>::draw() {
+void stratified_sampling<L>::draw(int N) {
+		update(N);
     double m, s, oldmean;
     for(int i=0;i<I;i++){
         m=0;