Merge pull request #106 from bridgerrholt/patch-1

Added get_weights and set_weights to fann_cpp.h
libfann · Jul 21, 2018 · b211dc3 · b211dc3
2 parents 2d2f707 + c82313b
commit b211dc3
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/include/fann_cpp.h b/src/include/fann_cpp.h
@@ -1909,6 +1909,36 @@ namespace FANN {
                 fann_set_weight(ann, from_neuron, to_neuron, weight);
             }
         }
+
+        /* Method: get_weights
+
+            Get all the network weights.
+
+            See also:
+                <fann_get_weights>
+
+            This function appears in FANN >= x.y.z
+        */
+        void get_weights(fann_type *weights) {
+            if (ann != NULL) {
+                fann_get_weights(ann, weights);
+            }
+        }
+
+        /* Method: set_weights
+        
+            Set network weights.
+            
+            See also:
+                <fann_set_weights>
+                
+            This function appears in FANN >= x.y.z
+        */
+        void set_weights(fann_type *weights) {
+            if (ann != NULL) {
+                fann_set_weights(ann, weights);
+            }
+        }
 
         /*********************************************************************/