braincog.base package

Subpackages

• braincog.base.brainarea package
  • Submodules
  • braincog.base.brainarea.BrainArea module
    • BrainArea
      • BrainArea.forward()
      • BrainArea.reset()
      • BrainArea.training
    • Feedback
      • Feedback.forward()
      • Feedback.reset()
      • Feedback.training
    • SelfConnectionArea
      • SelfConnectionArea.forward()
      • SelfConnectionArea.reset()
      • SelfConnectionArea.training
    • ThreePointForward
      • ThreePointForward.forward()
      • ThreePointForward.training
    • TwoInOneOut
      • TwoInOneOut.forward()
      • TwoInOneOut.training
  • braincog.base.brainarea.IPL module
    • IPLNet
      • IPLNet.UpdateWeight()
      • IPLNet.forward()
      • IPLNet.getweight()
      • IPLNet.reset()
      • IPLNet.training
  • braincog.base.brainarea.Insula module
    • InsulaNet
      • InsulaNet.UpdateWeight()
      • InsulaNet.forward()
      • InsulaNet.getweight()
      • InsulaNet.reset()
      • InsulaNet.training
  • braincog.base.brainarea.PFC module
    • PFC
      • PFC.forward()
      • PFC.reset()
    • dlPFC
      • dlPFC.training
  • braincog.base.brainarea.basalganglia module
    • basalganglia
      • basalganglia.UpdateWeight()
      • basalganglia.forward()
      • basalganglia.getmask()
      • basalganglia.getweight()
      • basalganglia.reset()
      • basalganglia.training
  • braincog.base.brainarea.dACC module
  • Module contents
    • BrainArea
      • BrainArea.forward()
      • BrainArea.reset()
      • BrainArea.training
    • Feedback
      • Feedback.forward()
      • Feedback.reset()
      • Feedback.training
    • IPLNet
      • IPLNet.UpdateWeight()
      • IPLNet.forward()
      • IPLNet.getweight()
      • IPLNet.reset()
      • IPLNet.training
    • InsulaNet
      • InsulaNet.UpdateWeight()
      • InsulaNet.forward()
      • InsulaNet.getweight()
      • InsulaNet.reset()
      • InsulaNet.training
    • PFC
      • PFC.forward()
      • PFC.reset()
    • SelfConnectionArea
      • SelfConnectionArea.forward()
      • SelfConnectionArea.reset()
      • SelfConnectionArea.training
    • ThreePointForward
      • ThreePointForward.forward()
      • ThreePointForward.training
    • TwoInOneOut
      • TwoInOneOut.forward()
      • TwoInOneOut.training
    • basalganglia
      • basalganglia.UpdateWeight()
      • basalganglia.forward()
      • basalganglia.getmask()
      • basalganglia.getweight()
      • basalganglia.reset()
      • basalganglia.training
    • dlPFC
      • dlPFC.training
• braincog.base.connection package
  • Submodules
  • braincog.base.connection.CustomLinear module
    • CustomLinear
      • CustomLinear.forward()
      • CustomLinear.training
      • CustomLinear.update()
  • braincog.base.connection.layer module
    • LIPool
      • LIPool.forward()
      • LIPool.reset()
      • LIPool.training
    • LayerNorm
      • LayerNorm.forward()
      • LayerNorm.training
    • NDropout
      • NDropout.create_mask()
      • NDropout.forward()
      • NDropout.n_reset()
      • NDropout.training
    • SMaxPool
      • SMaxPool.forward()
      • SMaxPool.reset()
      • SMaxPool.training
    • TEBN
      • TEBN.forward()
      • TEBN.training
    • ThresholdDependentBatchNorm2d
      • ThresholdDependentBatchNorm2d.affine
      • ThresholdDependentBatchNorm2d.eps
      • ThresholdDependentBatchNorm2d.forward()
      • ThresholdDependentBatchNorm2d.momentum
      • ThresholdDependentBatchNorm2d.num_features
      • ThresholdDependentBatchNorm2d.track_running_stats
    • VotingLayer
      • VotingLayer.forward()
      • VotingLayer.training
    • WTALayer
      • WTALayer.forward()
      • WTALayer.training
  • Module contents
    • CustomLinear
      • CustomLinear.forward()
      • CustomLinear.training
      • CustomLinear.update()
    • LIPool
      • LIPool.forward()
      • LIPool.reset()
      • LIPool.training
    • LayerNorm
      • LayerNorm.forward()
      • LayerNorm.training
    • NDropout
      • NDropout.create_mask()
      • NDropout.forward()
      • NDropout.n_reset()
      • NDropout.training
    • SMaxPool
      • SMaxPool.forward()
      • SMaxPool.reset()
      • SMaxPool.training
    • ThresholdDependentBatchNorm2d
      • ThresholdDependentBatchNorm2d.affine
      • ThresholdDependentBatchNorm2d.eps
      • ThresholdDependentBatchNorm2d.forward()
      • ThresholdDependentBatchNorm2d.momentum
      • ThresholdDependentBatchNorm2d.num_batches_tracked
      • ThresholdDependentBatchNorm2d.num_features
      • ThresholdDependentBatchNorm2d.running_mean
      • ThresholdDependentBatchNorm2d.running_var
      • ThresholdDependentBatchNorm2d.track_running_stats
      • ThresholdDependentBatchNorm2d.training
    • VotingLayer
      • VotingLayer.forward()
      • VotingLayer.training
    • WTALayer
      • WTALayer.forward()
      • WTALayer.training
• braincog.base.conversion package
  • Submodules
  • braincog.base.conversion.convertor module
    • Convertor
      • Convertor.forward()
      • Convertor.get_percentile()
      • Convertor.register_hook()
      • Convertor.replace_for_spike()
      • Convertor.training
    • HookScale
      • HookScale.forward()
      • HookScale.training
    • Hookoutput
      • Hookoutput.forward()
      • Hookoutput.training
    • SNode
      • SNode.forward()
      • SNode.hard_reset()
      • SNode.reset()
      • SNode.soft_reset()
      • SNode.training
    • Scale
      • Scale.forward()
      • Scale.training
    • reset()
  • braincog.base.conversion.merge module
    • merge()
    • mergeConvBN()
  • Module contents
    • Convertor
      • Convertor.forward()
      • Convertor.get_percentile()
      • Convertor.register_hook()
      • Convertor.replace_for_spike()
      • Convertor.training
    • HookScale
      • HookScale.forward()
      • HookScale.training
    • Hookoutput
      • Hookoutput.forward()
      • Hookoutput.training
    • SNode
      • SNode.forward()
      • SNode.hard_reset()
      • SNode.reset()
      • SNode.soft_reset()
      • SNode.training
    • Scale
      • Scale.forward()
      • Scale.training
    • merge()
    • mergeConvBN()
• braincog.base.encoder package
  • Submodules
  • braincog.base.encoder.encoder module
    • AutoEncoder
      • AutoEncoder.forward()
      • AutoEncoder.training
    • Encoder
      • Encoder.auto()
      • Encoder.delete()
      • Encoder.direct()
      • Encoder.forward()
      • Encoder.phase()
      • Encoder.rate()
      • Encoder.shift()
      • Encoder.training
      • Encoder.ttfs()
  • braincog.base.encoder.population_coding module
    • PEncoder
      • PEncoder.forward()
      • PEncoder.population_time()
      • PEncoder.population_voltage()
      • PEncoder.training
  • braincog.base.encoder.qs_coding module
    • QSEncoder
      • QSEncoder.noise_trans()
      • QSEncoder.reverse_pixels()
      • QSEncoder.shift_trans()
  • Module contents
    • Encoder
      • Encoder.auto()
      • Encoder.delete()
      • Encoder.direct()
      • Encoder.forward()
      • Encoder.phase()
      • Encoder.rate()
      • Encoder.shift()
      • Encoder.training
      • Encoder.ttfs()
    • PEncoder
      • PEncoder.forward()
      • PEncoder.population_time()
      • PEncoder.population_voltage()
      • PEncoder.training
    • QSEncoder
      • QSEncoder.noise_trans()
      • QSEncoder.reverse_pixels()
      • QSEncoder.shift_trans()
• braincog.base.learningrule package
  • Submodules
  • braincog.base.learningrule.BCM module
    • BCM
      • BCM.cfunc()
      • BCM.forward()
      • BCM.reset()
      • BCM.training
  • braincog.base.learningrule.Hebb module
    • Hebb
      • Hebb.forward()
      • Hebb.reset()
      • Hebb.training
  • braincog.base.learningrule.RSTDP module
    • RSTDP
      • RSTDP.cal_trace()
      • RSTDP.forward()
      • RSTDP.reset()
      • RSTDP.training
  • braincog.base.learningrule.STDP module
    • FullSTDP
      • FullSTDP.cal_tracein()
      • FullSTDP.cal_traceout()
      • FullSTDP.forward()
      • FullSTDP.reset()
      • FullSTDP.training
    • LTD
      • LTD.cal_trace()
      • LTD.forward()
      • LTD.reset()
      • LTD.training
    • LTP
      • LTP.training
    • MutliInputSTDP
      • MutliInputSTDP.cal_trace()
      • MutliInputSTDP.forward()
      • MutliInputSTDP.reset()
      • MutliInputSTDP.training
    • STDP
      • STDP.cal_trace()
      • STDP.forward()
      • STDP.reset()
      • STDP.training
  • braincog.base.learningrule.STP module
    • short_time
      • short_time.IDderiv()
      • short_time.set_gsyn()
      • short_time.syndepr()
      • short_time.update()
  • Module contents
    • BCM
      • BCM.cfunc()
      • BCM.forward()
      • BCM.reset()
      • BCM.training
    • FullSTDP
      • FullSTDP.cal_tracein()
      • FullSTDP.cal_traceout()
      • FullSTDP.forward()
      • FullSTDP.reset()
      • FullSTDP.training
    • Hebb
      • Hebb.forward()
      • Hebb.reset()
      • Hebb.training
    • LTD
      • LTD.cal_trace()
      • LTD.forward()
      • LTD.reset()
      • LTD.training
    • LTP
      • LTP.training
    • MutliInputSTDP
      • MutliInputSTDP.cal_trace()
      • MutliInputSTDP.forward()
      • MutliInputSTDP.reset()
      • MutliInputSTDP.training
    • RSTDP
      • RSTDP.cal_trace()
      • RSTDP.forward()
      • RSTDP.reset()
      • RSTDP.training
    • STDP
      • STDP.cal_trace()
      • STDP.forward()
      • STDP.reset()
      • STDP.training
    • short_time
      • short_time.IDderiv()
      • short_time.set_gsyn()
      • short_time.syndepr()
      • short_time.update()
• braincog.base.node package
  • Submodules
  • braincog.base.node.node module
    • BackEINode
      • BackEINode.calc_spike()
      • BackEINode.integral()
      • BackEINode.n_reset()
      • BackEINode.training
    • BaseMCNode
      • BaseMCNode.calc_spike()
      • BaseMCNode.forward()
      • BaseMCNode.get_n_fire_rate()
      • BaseMCNode.integral()
      • BaseMCNode.n_reset()
      • BaseMCNode.set_n_threshold()
      • BaseMCNode.set_n_warm_up()
      • BaseMCNode.training
    • BaseNode
      • BaseNode.calc_spike()
      • BaseNode.forward()
      • BaseNode.get_n_attr()
      • BaseNode.get_thres()
      • BaseNode.integral()
      • BaseNode.n_reset()
      • BaseNode.rearrange2node()
      • BaseNode.rearrange2op()
      • BaseNode.set_n_tau()
      • BaseNode.set_n_threshold()
      • BaseNode.set_n_warm_up()
      • BaseNode.training
    • BiasLIFNode
      • BiasLIFNode.calc_spike()
      • BiasLIFNode.integral()
      • BiasLIFNode.training
    • BiasPLIFNode
      • BiasPLIFNode.calc_spike()
      • BiasPLIFNode.integral()
      • BiasPLIFNode.training
    • BiasReLUNode
      • BiasReLUNode.calc_spike()
      • BiasReLUNode.forward()
      • BiasReLUNode.training
    • BinaryNode
      • BinaryNode.calc_spike()
      • BinaryNode.forward()
      • BinaryNode.training
    • CTIzhNode
      • CTIzhNode.calc_spike()
      • CTIzhNode.integral()
      • CTIzhNode.spreadMarkPostNeurons()
      • CTIzhNode.training
    • DGLIFNode
      • DGLIFNode.calc_spike()
      • DGLIFNode.integral()
      • DGLIFNode.training
    • DoubleSidePLIFNode
      • DoubleSidePLIFNode.calc_spike()
      • DoubleSidePLIFNode.training
    • HTDGLIFNode
      • HTDGLIFNode.calc_spike()
      • HTDGLIFNode.forward()
      • HTDGLIFNode.training
    • IFNode
      • IFNode.calc_spike()
      • IFNode.integral()
      • IFNode.training
    • IRNode
      • IRNode.calc_spike()
      • IRNode.forward()
      • IRNode.training
    • IzhNode
      • IzhNode.calc_spike()
      • IzhNode.d
      • IzhNode.integral()
      • IzhNode.n_reset()
      • IzhNode.training
    • IzhNodeMU
      • IzhNodeMU.calc_spike()
      • IzhNodeMU.integral()
      • IzhNodeMU.n_reset()
      • IzhNodeMU.requires_activation()
      • IzhNodeMU.training
    • LIAFNode
      • LIAFNode.calc_spike()
      • LIAFNode.integral()
      • LIAFNode.training
    • LIFNode
      • LIFNode.calc_spike()
      • LIFNode.integral()
      • LIFNode.training
    • LIFSTDPNode
      • LIFSTDPNode.calc_spike()
      • LIFSTDPNode.integral()
      • LIFSTDPNode.requires_activation()
      • LIFSTDPNode.training
    • NoiseLIFNode
      • NoiseLIFNode.integral()
      • NoiseLIFNode.training
    • NoisePLIFNode
      • NoisePLIFNode.integral()
      • NoisePLIFNode.training
    • PLIFNode
      • PLIFNode.calc_spike()
      • PLIFNode.integral()
      • PLIFNode.training
    • ReLUNode
      • ReLUNode.calc_spike()
      • ReLUNode.forward()
      • ReLUNode.training
    • SimHHNode
      • SimHHNode.calc_spike()
      • SimHHNode.forward()
      • SimHHNode.integral()
      • SimHHNode.n_reset()
      • SimHHNode.requires_activation()
      • SimHHNode.training
    • ThreeCompNode
      • ThreeCompNode.calc_spike()
      • ThreeCompNode.integral()
      • ThreeCompNode.training
    • aEIF
      • aEIF.aEIFNode()
      • aEIF.calc_spike()
      • aEIF.training
  • Module contents
• braincog.base.strategy package
  • Submodules
  • braincog.base.strategy.LateralInhibition module
    • LateralInhibition
      • LateralInhibition.forward()
      • LateralInhibition.training
  • braincog.base.strategy.surrogate module
    • AtanGrad
      • AtanGrad.act_fun()
      • AtanGrad.training
    • BackEIGateGrad
      • BackEIGateGrad.act_fun()
      • BackEIGateGrad.training
    • EIGrad
      • EIGrad.act_fun()
      • EIGrad.training
    • GateGrad
      • GateGrad.act_fun()
      • GateGrad.training
    • QGateGrad
      • QGateGrad.act_fun()
      • QGateGrad.training
    • ReLUGrad
      • ReLUGrad.act_fun()
      • ReLUGrad.training
    • STDPGrad
      • STDPGrad.act_fun()
      • STDPGrad.training
    • SigmoidGrad
      • SigmoidGrad.act_fun()
      • SigmoidGrad.training
    • SurrogateFunctionBase
      • SurrogateFunctionBase.act_fun()
      • SurrogateFunctionBase.forward()
      • SurrogateFunctionBase.training
    • atan
      • atan.backward()
      • atan.forward()
    • backeigate
      • backeigate.backward()
      • backeigate.forward()
    • ei
      • ei.backward()
      • ei.forward()
    • gate
      • gate.backward()
      • gate.forward()
    • heaviside()
    • quadratic_gate
      • quadratic_gate.backward()
      • quadratic_gate.forward()
    • relu_like
      • relu_like.backward()
      • relu_like.forward()
    • sigmoid
      • sigmoid.backward()
      • sigmoid.forward()
    • stdp
      • stdp.backward()
      • stdp.forward()
    • straight_through_estimator
      • straight_through_estimator.backward()
      • straight_through_estimator.forward()
  • Module contents
• braincog.base.utils package
  • Submodules
  • braincog.base.utils.criterions module
    • MixLoss
      • MixLoss.forward()
      • MixLoss.training
    • UnilateralMse
      • UnilateralMse.forward()
      • UnilateralMse.training
  • braincog.base.utils.visualization module
    • plot_confusion_matrix()
    • plot_mem_distribution()
    • plot_tsne()
    • plot_tsne_3d()
    • spike_rate_vis()
    • spike_rate_vis_1d()
  • Module contents
    • MixLoss
      • MixLoss.forward()
      • MixLoss.training
    • UnilateralMse
      • UnilateralMse.forward()
      • UnilateralMse.training
    • plot_confusion_matrix()
    • plot_tsne()
    • plot_tsne_3d()

Module contents