Example

Process

1. Initialise configurations
2. Define the sequence. Alternatively, read this from a file or from the terminal. Your choice.
3. Initialise a region
4. On each input value
   1. encode the value as SDR
   2. pass the SDR to the spatial pooler along with the region
   3. pass the updated region to the temporal pooler. The region updated by this pooler is passed to the spatial pooler on the next input.

You can

1. Store/display the region as you want by defining your own parser or using the displayRegion method from SDR.Region.Utils. In the future, there might be more functions that can convert a Region to an SDR or other forms; depending on use cases.

Example

import Control.Lens ((^.)) -- Used for record access
import System.Exit (exitFailure) 
import System.Random (mkStdGen)
import System.Random.Shuffle ()
import SRC.Encoder.Config
import SRC.Encoder.Numeric
  ( encode,
    getRange,
  )
import SRC.HTM.Config
import SRC.HTM.HTM
import SRC.Region.Region
import SRC.Package
import SRC.SDR


initEncoderConfig :: EncoderConfig
initEncoderConfig =
  EncoderConfig
    { _encoderType = Numeric,
      _minVal = 0,
      _maxVal = 100,
      _buckets = 50,
      _bitsPerBucket = 20
    }

initRegionConfig :: RegionConfig
initRegionConfig =
  RegionConfig
    { _nrOfColumns = 100,
      _nrOfCellsPerColumn = 2,
      _initNrOfFeedForwardSynpases = 20,
      _nrOfSynapsesPerSegment = 4,
      _mappingType = Random,
      _initConnectionStrength = 0.7,
      _mvWindow = 3,
      _initRad = 5
    }

initHTMConfig :: HTMConfig
initHTMConfig =
  HTMConfig
    { _spatialConfig =
        SpatialConfig
          { _overlapThreshold = 2,
            _mop = 0.2,
            _proxSynConInc = 0.2,
            _proxSynConDec = 0.2,
            _pConthresh = 0.2,
            _colActLev = 1
          },
      _temporalConfig =
        TemporalConfig
          { _targetDensity = 0.3,
            _boostStrength = 0.3,
            _connectedPermenance = 0.5,
            _activationThreshold = 2,
            _predictedDecrement = 0.1,
            _permanenceIncrement = 0.2,
            _permanenceDecrement = 0.2,
            _learningThreshold = 2,
            _learningEnabled = True
          }
    }

initConfigs :: IO Package
initConfigs = do
  let s = initEncoderConfig
  case getRange s of
    Just a -> do
      let r = initRegionConfig
      let h = initHTMConfig
      return $
        Package
          { _conH = h,
            _conR = r,
            _conS = s,
            _value = SDR [] a,
            _randomGenerator = mkStdGen 12
          }
    Nothing -> exitFailure

main :: IO ()
main = do
  -- Configurations
  package <- initConfigs

  -- Initialisation a region
  region <- initRegion (package^. conS) (package^.conR)

  -- Input data
  -- let seqData = [20, 20, 20, 20, 20, 20, 20, 20, 77, 77, 77, 77, 77, 77, 77, 30, 30, 30, 30]

  -- Learning/Predicting, i.e. spatial and temporal poolers
  compute seqData package region


-- | Apply HTM on a sequence of input data.
compute :: [Int] -> Package -> Region -> IO ()
compute [] _ _ = do
  putStrLn "encoding completed"

compute (x : xs) p region = do
  putStrLn $ "----- next encoding. val: " ++ show x
  let encodedSDR = encode (p^.conS) x
  case encodedSDR of
    Just val ->
      do
        print val
        -- Spatial Encoding
        let regionSpat = spatialPooler p {_value = val} region -- Updates columnState among other variables

        -- Temporal encoding
        regionTemp <- temporalPooler p regionSpat -- Updates cellState among other variables
        putStrLn $ displayRegion regionTemp -- This shows the state of all cells
        
        compute xs p $ switch regionTemp
    Nothing ->
      print "Invalid value"