hotfix

micromlgen/templates/compute_kernels.jinja

@@ -1,4 +1,4 @@
-{% if platform == 'attiny' %}
+{% if isAttiny %}
     float w[{{ FEATURES_DIM }}];
 {% endif %}
 

micromlgen/templates/kernel_function.jinja

@@ -3,7 +3,7 @@
  * Kernel type: {{ KERNEL_TYPE }}
  */
 {% if isAttiny %}
-double compute_kernel(double x[{{ FEATURES_DIM }}], double w[{{ FEATURES_DIM }}]) {
+double compute_kernel(double x[{{ FEATURES_DIM }}], float w[{{ FEATURES_DIM }}]) {
     double kernel = 0.0;
 {% else %}
 double compute_kernel(double x[{{ FEATURES_DIM }}], ...) {
@@ -15,9 +15,9 @@ double compute_kernel(double x[{{ FEATURES_DIM }}], ...) {
 
     for (uint16_t i = 0; i < {{ FEATURES_DIM }}; i++)
         {% if KERNEL_TYPE in ['linear', 'poly', 'sigmoid'] %}
-            kernel += x[i] * {% if isAttiny %} w[i] {% else %} va_arg(w, double), 2) {% endif %};
+            kernel += x[i] * {% if isAttiny %} w[i] {% else %} va_arg(w, double) {% endif %};
         {% elif KERNEL_TYPE == 'rbf' %}
-            kernel += pow(x[i] - {% if isAttiny %} w[i] {% else %} va_arg(w, double), 2) {% endif %};
+            kernel += pow(x[i] - {% if isAttiny %} w[i] {% else %} va_arg(w, double) {% endif %}, 2);
         {% else %}
             #error "UNKNOWN KERNEL {{ kernel }}";
         {% endif %}

publish

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+# user agrimag
+
 #git push origin master -f
 rm -rf dist/*
 python setup.py sdist

setup.py

@@ -2,13 +2,13 @@
 setup(
   name = 'micromlgen',
   packages = ['micromlgen'],
-  version = '0.6',
+  version = '0.7',
   license='MIT',
   description = 'Generate C code for microcontrollers from Python\'s sklearn classifiers',
   author = 'Simone Salerno',
   author_email = '[email protected]',
   url = 'https://github.com/agrimagsrl/micromlgen',
-  download_url = 'https://github.com/agrimagsrl/micromlgen/archive/v_06.tar.gz',
+  download_url = 'https://github.com/agrimagsrl/micromlgen/archive/v_07.tar.gz',
   keywords = ['ML', 'microcontrollers', 'sklearn', 'machine learning'],
   install_requires=[
     'jinja2',