Merge pull request #40 from alexander-yu/alex

Add weights for each iteration in demo2

Author: Alex Yu

demo/demo2.neo

@@ -11,6 +11,7 @@ int sum(matrix<int> X) {
 
 matrix<int> train(matrix<int> X, matrix<int> y) {
     create matrix<int> weights[1][cols(X)];
+    auto iter_weights = weights[:, :];
     auto has_errors = True;
 
     while (has_errors) {
@@ -39,10 +40,19 @@ matrix<int> train(matrix<int> X, matrix<int> y) {
 
         if (count == 0) {
             has_errors = False;
+        } else {
+            auto weights_copy = weights[:, :];
+            auto new_iter_weights = append(iter_weights, weights_copy);
+
+            free(weights_copy);
+            free(iter_weights);
+
+            iter_weights = new_iter_weights;
         }
     }
 
-    return weights;
+    free(weights);
+    return iter_weights;
 }
 
 void main() {

demo/weights2.mats

@@ -1,2 +1,21 @@
-1,3,int
+20,3,int
+0, 0, 0
+3, 8, 3
+6, 3, 9
+10, 12, 6
+12, 6, 1
+15, 7, 9
+16, 5, 2
+18, 5, -1
+21, 6, 7
+23, 3, 8
+24, -1, -2
+26, 2, 7
+28, 3, 3
+30, 3, 6
+32, 0, 7
+34, 1, 3
+36, 1, 6
+38, 2, 2
+40, 2, 5
 39, -5, -2