ajout fonction estimator

1 files changed, 15 insertions, 2 deletions

diff --git a/stratified_sampling.hpp b/stratified_sampling.hpp
index ca96a3c..e9e1823 100644
--- a/stratified_sampling.hpp
+++ b/stratified_sampling.hpp
@@ -54,7 +54,7 @@ struct stratified_sampling {
     vector<double> get_mean();
     vector<double> get_var();
     void print_mean();
-    //double estimator();
+    pair<double,double> estimator();
 private:
     vector<double> p;
     vector<int> M;
@@ -147,6 +147,19 @@ void stratified_sampling<L>::print_mean() {
         cout<<mean[i]<<"\t";
 			}
 			cout<<endl;
-		}
+};
+
+template <typename L>
+pair<double,double> stratified_sampling<L>::estimator() {
+	double est_mean;
+	double est_std;
+	for (int i=0; i<I; i++) {
+		est_mean += mean[i]*p[i];
+		est_std += sqrt(sigma2[i])*p[i];
+	}
+	return {est_mean, est_std};
+}
+	
+