add GaussianNB

micromlgen/templates/dot.jinja

@@ -16,7 +16,7 @@ float dot(float *x, {{ signature }}) {
     float dot = 0.0;
 
     for (uint16_t i = 0; i < {{ dimension }}; i++) {
-        const float wi = {{ wi }}
+        const float wi = {{ wi }};
         dot += {{ expr }};
     }
 

micromlgen/templates/gaussiannb/gaussiannb.jinja

@@ -0,0 +1,40 @@
+#pragma once
+
+namespace Eloquent {
+    namespace ML {
+        namespace Port {
+
+            class {{ classname }} {
+                public:
+
+                    /**
+                     * Predict class for features vector
+                     */
+                    int predict(float *x) {
+                        float votes[{{ classes|length }}] = { 0.0f };
+
+                        {% include 'gaussiannb/vote.jinja' %}
+                        {% include 'vote.jinja' %}
+                    }
+
+                    {% include 'classmap.jinja' %}
+
+                protected:
+
+                    /**
+                     * Compute gaussian value
+                     */
+                    float gauss(float *x, float *theta, float *sigma) {
+                        float gauss = 0.0f;
+
+                        for (uint16_t i = 0; i < {{ theta[0]|length }}; i++) {
+                            gauss += log(sigma[i]);
+                            gauss += pow(x[i] - theta[i], 2) / sigma[i];
+                        }
+
+                        return gauss;
+                    }
+            };
+        }
+    }
+}

micromlgen/templates/gaussiannb/vote.jinja

@@ -0,0 +1,8 @@
+float theta[{{ theta[0]|length }}] = { 0 };
+float sigma[{{ sigma[0]|length }}] = { 0 };
+
+{% for i, (t, s) in f.enumerate(f.zip(theta, sigma)) %}
+    {% for j, tj in f.enumerate(t) %}theta[{{ j }}] = {{ f.round(tj) }}; {% endfor %}
+    {% for j, sj in f.enumerate(s) %}sigma[{{ j }}] = {{ f.round(sj) }}; {% endfor %}
+    votes[{{ i }}] = {{ f.round(prior[i]) }} - gauss(x, theta, sigma);
+{% endfor %}

micromlgen/templates/vote.jinja

@@ -1,6 +1,6 @@
 // return argmax of votes
 uint8_t classIdx = 0;
-uint8_t maxVotes = votes[0];
+float maxVotes = votes[0];
 
 for (uint8_t i = 1; i < {{ n_classes }}; i++) {
     if (votes[i] > maxVotes) {