Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numba as nb
- import numpy as np
- @nb.jit
- def c(nbr):
- return nbr
- class Axon:
- def __init__(self):
- self.c = Fifo()
- print self.c.res
- class Fifo:
- def __init__(self):
- fifos = []
- for i in range(3):
- fifos.append(BinaryFifo(i))
- res = 0
- for j in range(3):
- res += fifos[j].bb
- self.res = res
- class BinaryFifo:
- def __init__(self,nbr):
- self.bb = c(nbr)
- if __name__ == "__main__":
- Axon()
- @jit
- def get_post_spike_energy_jit(positive_weight,negative_weight,eligibility,betas):
- pos_w_energy = np.dot(positive_weight,eligibility)
- neg_w_energy = np.dot(negative_weight,betas)
- tot_energy = neg_w_energy - pos_w_energy
- return tot_energy
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement