fix kernel function

agrimagsrl · Dec 21, 2019 · 1dcb1ef · 1dcb1ef
1 parent 98de2d0
commit 1dcb1ef
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 24 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 venv
 .idea
+tests
diff --git a/micromlgen/__pycache__/__init__.cpython-36.pyc b/micromlgen/__pycache__/__init__.cpython-36.pyc
diff --git a/micromlgen/__pycache__/micromlgen.cpython-36.pyc b/micromlgen/__pycache__/micromlgen.cpython-36.pyc
diff --git a/micromlgen/micromlgen_test.py b/micromlgen/micromlgen_test.py
diff --git a/micromlgen/templates/kernel_function.jinja b/micromlgen/templates/kernel_function.jinja
@@ -18,11 +18,11 @@ double compute_kernel(double x[{{ FEATURES_DIM }}], ...) {
         {% endif %}
 
     {% if KERNEL_TYPE == 'poly' %}
-        kernel = pow((KERNEL_GAMMA * kernel) + KERNEL_COEF, KERNEL_DEGREE);
+        kernel = pow(({{ KERNEL_GAMMA }} * kernel) + {{ KERNEL_COEF }}, {{ KERNEL_DEGREE }});
     {% elif KERNEL_TYPE == 'rbf' %}
         kernel = exp(-{{ KERNEL_GAMMA }} * kernel);
     {% elif KERNEL_TYPE == 'sigmoid' %}
-        kernel = sigmoid((KERNEL_GAMMA * kernel) + KERNEL_COEF);
+        kernel = sigmoid(({{ KERNEL_GAMMA }} * kernel) + {{ KERNEL_COEF }});
     {% endif %}
 
     return kernel;
-Original file line number
+Diff line change
@@ -1,2 +1,3 @@
     venv
     .idea
+    tests