add new constructor in gaussian kernel

shogun-toolbox · May 31, 2016 · 464c49c · 464c49c
1 parent 3659533
commit 464c49c
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/src/shogun/kernel/GaussianKernel.cpp b/src/shogun/kernel/GaussianKernel.cpp
@@ -36,6 +36,12 @@ CGaussianKernel::CGaussianKernel() : CDotKernel()
 	init();
 }
 
+CGaussianKernel::CGaussianKernel(float64_t w) : CDotKernel()
+{
+	init();
+	set_width(w);
+}
+
 CGaussianKernel::CGaussianKernel(int32_t size, float64_t w) : CDotKernel(size)
 {
 	init();

diff --git a/src/shogun/kernel/GaussianKernel.h b/src/shogun/kernel/GaussianKernel.h
@@ -42,6 +42,12 @@ class CGaussianKernel: public CDotKernel
 		/** default constructor */
 		CGaussianKernel();
 
+		/** constructor
+		 *
+		 * @param width width
+		 */
+		CGaussianKernel(float64_t width);
+
 		/** constructor
 		 *
 		 * @param size cache size

diff --git a/tests/unit/kernel/Kernel_unittest.cc b/tests/unit/kernel/Kernel_unittest.cc
@@ -448,3 +448,11 @@ TEST(Kernel, row_col_wise_sum_block_no_diag)
 	// cleanup
 	SG_UNREF(kernel);
 }
+
+TEST(Kernel, gaussian_kernel_width_constructor)
+{
+	float64_t width = 5;
+	CGaussianKernel* kernel=new CGaussianKernel(width);
+	EXPECT_EQ(kernel->get_cache_size(), 10);
+	EXPECT_EQ(kernel->get_width(), width);
+}