TM Connections integration

bindings/py/cpp_src/bindings/algorithms/py_Connections.cpp

@@ -49,10 +49,14 @@ R"(Compatibility Warning: This classes API is unstable and may change without wa
         [](const Connections &self) { return self.getConnectedThreshold(); });
 
     py_Connections.def("createSegment", &Connections::createSegment,
-        py::arg("cell"));
+        py::arg("cell"),
+	py::arg("maxSegmentsPerCell") = 0
+	);
 
     py_Connections.def("destroySegment", &Connections::destroySegment);
 
+    py_Connections.def("iteration", &Connections::iteration);
+
     py_Connections.def("createSynapse", &Connections::createSynapse,
         py::arg("segment"),
         py::arg("presynaticCell"),
@@ -93,24 +97,26 @@ R"(Compatibility Warning: This classes API is unstable and may change without wa
     py_Connections.def("reset", &Connections::reset);
 
     py_Connections.def("computeActivity",
-        [](Connections &self, SDR &activePresynapticCells) {
+        [](Connections &self, SDR &activePresynapticCells, bool learn=true) {
             // Allocate buffer to return & make a python destructor object for it.
             auto activeConnectedSynapses =
                 new std::vector<SynapseIdx>( self.segmentFlatListLength(), 0u );
             auto destructor = py::capsule( activeConnectedSynapses,
                 [](void *dataPtr) {
                     delete reinterpret_cast<std::vector<SynapseIdx>*>(dataPtr); });
-            // Call the C++ method.
-            self.computeActivity(*activeConnectedSynapses, activePresynapticCells.getSparse());
-            // Wrap vector in numpy array.
+
+	    // Call the C++ method.
+            self.computeActivity(*activeConnectedSynapses, activePresynapticCells.getSparse(), learn);
+
+	    // Wrap vector in numpy array.
             return py::array(activeConnectedSynapses->size(),
                              activeConnectedSynapses->data(),
                              destructor);
         },
 R"(Returns numActiveConnectedSynapsesForSegment)");
 
     py_Connections.def("computeActivityFull",
-        [](Connections &self, SDR &activePresynapticCells) {
+        [](Connections &self, SDR &activePresynapticCells, bool learn=true) {
             // Allocate buffer to return & make a python destructor object for it.
             auto activeConnectedSynapses =
                 new std::vector<SynapseIdx>( self.segmentFlatListLength(), 0u );
@@ -123,9 +129,13 @@ R"(Returns numActiveConnectedSynapsesForSegment)");
             auto potentialDestructor = py::capsule( activePotentialSynapses,
                 [](void *dataPtr) {
                     delete reinterpret_cast<std::vector<SynapseIdx>*>(dataPtr); });
-            // Call the C++ method.
-            self.computeActivity(*activeConnectedSynapses, *activePotentialSynapses,
-                                            activePresynapticCells.getSparse());
+
+	    // Call the C++ method.
+            self.computeActivity(*activeConnectedSynapses, 
+			         *activePotentialSynapses,
+                                 activePresynapticCells.getSparse(),
+				 learn);
+
             // Wrap vector in numpy array.
             return py::make_tuple(
                     py::array(activeConnectedSynapses->size(),

src/examples/hotgym/HelloSPTP.cpp

@@ -29,6 +29,7 @@
 #include "htm/types/Sdr.hpp"
 #include "htm/utils/Random.hpp"
 #include "htm/utils/MovingAverage.hpp"
+#include "htm/utils/SdrMetrics.hpp"
 
 namespace examples {
 
@@ -79,6 +80,13 @@ Real64 BenchmarkHotgym::run(UInt EPOCHS, bool useSPlocal, bool useSPglobal, bool
   SDR outTM(spGlobal.getColumnDimensions()); 
   Real an = 0.0f, anLikely = 0.0f; //for anomaly:
   MovingAverage avgAnom10(1000); //chose the window large enough so there's (some) periodicity in the patter, so TM can learn something
+
+  //metrics
+  Metrics statsInput(input, 1000);
+  Metrics statsSPlocal(outSPlocal, 1000);
+  Metrics statsSPglobal(outSPglobal, 1000);
+  Metrics statsTM(outTM, 1000);
+
   /*
    * For example: fn = sin(x) -> periodic >= 2Pi ~ 6.3 && x+=0.01 -> 630 steps to 1st period -> window >= 630
    */
@@ -147,13 +155,26 @@ Real64 BenchmarkHotgym::run(UInt EPOCHS, bool useSPlocal, bool useSPglobal, bool
     if (e == EPOCHS - 1) {
       tAll.stop();
 
+      //print connections stats
+      cout << "\nInput :\n" << statsInput
+	   << "\nSP(local) " << spLocal.connections
+	   << "\nSP(local) " << statsSPlocal
+           << "\nSP(global) " << spGlobal.connections
+	   << "\nSP(global) " << statsSPglobal
+           << "\nTM " << tm.connections 
+	   << "\nTM " << statsTM
+	   << "\n";
+
+      // output values
       cout << "Epoch = " << e << endl;
       cout << "Anomaly = " << an << endl;
       cout << "Anomaly (avg) = " << avgAnom10.getCurrentAvg() << endl;
       cout << "Anomaly (Likelihood) = " << anLikely << endl;
       cout << "SP (g)= " << outSP << endl;
       cout << "SP (l)= " << outSPlocal <<endl;
       cout << "TM= " << outTM << endl;
+
+      //timers
       cout << "==============TIMERS============" << endl;
       cout << "Init:\t" << tInit.getElapsed() << endl;
       cout << "Random:\t" << tRng.getElapsed() << endl;