first working method

shogun-toolbox · Jul 22, 2012 · f1232bc · f1232bc
1 parent 9f65805
commit f1232bc
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/shogun/features/DataGenerator.cpp b/src/shogun/features/DataGenerator.cpp
@@ -8,6 +8,7 @@
  */
 
 #include <shogun/features/DataGenerator.h>
+#include <shogun/mathematics/Math.h>
 
 using namespace shogun;
 
@@ -24,3 +25,19 @@ CDataGenerator::~CDataGenerator()
 void CDataGenerator::init()
 {
 }
+
+SGMatrix<float64_t> CDataGenerator::generate_mean_data(index_t m,
+		index_t dim, float64_t mean_shift)
+{
+	SGMatrix<float64_t> result(dim, m);
+
+	for (index_t i=0; i<m; ++i)
+	{
+		for (index_t j=0; j<dim; ++j)
+			result(j,i)=CMath::randn_double();
+
+		result(0,i)+=mean_shift;
+	}
+
+	return result;
+}
diff --git a/src/shogun/features/DataGenerator.h b/src/shogun/features/DataGenerator.h
@@ -11,6 +11,7 @@
 #define __DATAGENERATOR_H_
 
 #include <shogun/base/SGObject.h>
+#include <shogun/lib/SGMatrix.h>
 
 namespace shogun
 {
@@ -25,6 +26,18 @@ class CDataGenerator: public CSGObject
 
 	virtual ~CDataGenerator();
 
+	/** produces a set of vectors, where each dimension is standard normally
+	 * distributed and the mean of the first dimension is shifted by a
+	 * provided value
+	 *
+	 * @param m number of samples to generate
+	 * @param dim dimension of generated samples
+	 * @param mean_shift is added mean of first dimension
+	 * @return matrix with samples
+	 */
+	SGMatrix<float64_t> generate_mean_data(index_t m, index_t dim,
+			float64_t mean_shift);
+
 	inline virtual const char* get_name() const { return "DataGenerator"; }
 
 private: