Scribd
Upload
5 views

problem_py

Uploaded by

Suchandana Saha Joyguru
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
5 views

problem_py

Uploaded by

Suchandana Saha Joyguru
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
You are on page 1/ 4

# BIC F20 Assignment 2 Problem 1

# Written by Tristan Kilper

import numpy as np

# Sample feature set + labels


feature_set = np.array([[66.5,66.5],[67,66.5],[66.5,67],[67,67]])
labels1 = np.array([66.5,66.5,66.5,67])
labels1 = labels1.reshape(4,1)

# random seed
np.random.seed(42)

# input neuron x

def in_x(Ix, t):


Cm = 1.6
Rm = 1.5
input_I = Ix
Vm = 2
thold = 0
Vr = -65

Vx = 66

def out_V(Vcur, t):


return ((-1 * (Vcur - Vr) + input_I*t)/Cm) - (Vm*t)/(Rm*Cm)

time = t # ms
cur_t = 0
cur_out = Vr

while cur_t <= time:


print('t: ' + str(cur_t) + ' V: ' + str(cur_out))
cur_out += out_V(cur_out, 1)
if cur_out >= thold:
print('[spike!]')
cur_out = Vr
Vx += .5
cur_t += 1

return Vx

# input neuron y

def in_y(Iy, t):


Cm = 1.6
Rm = 1.5
input_I = Iy
Vm = 2
thold = 0
Vr = -65

Vy = 66

def out_V(Vcur, t):


return ((-1 * (Vcur - Vr) + input_I*t)/Cm) - (Vm*t)/(Rm*Cm)

time = t # ms
cur_t = 0
cur_out = Vr

while cur_t <= time:


print('t: ' + str(cur_t) + ' V: ' + str(cur_out))
cur_out += out_V(cur_out, 1)
if cur_out >= thold:
print('[spike!]')
cur_out = Vr
Vy += .5
cur_t += 1

return Vy

# output neuron z

def out_z(Iz, t):


Cm = 1.6
Rm = 1.5
input_I = Iz
Vm = 2
thold = 0
Vr = -65

Vz = 66

def out_V(Vcur, t):


return ((-1 * (Vcur - Vr) + input_I*t)/Cm) - (Vm*t)/(Rm*Cm)

time = t # ms
cur_t = 0
cur_out = Vr

while cur_t <= time:


print('t: ' + str(cur_t) + ' V: ' + str(cur_out))
cur_out += out_V(cur_out, 1)
if cur_out >= thold:
print('[spike!]')
cur_out = Vr
Vz += .5
cur_t += 1

return Vz

# learning rules
a1 = 0.0000005
a2 = 0.0000001
a3 = 0.0000001

def d_w1(x,z):
return (a1*x*z) - (a2*x) - (a3*z)

b1 = 0.0000005
b2 = 0.0000001
b3 = 0.0000001
def d_w2(y,z):
return (b1*y*z) - (b2*y) - (b3*z)

# translate spike train to inputs


def readSpikeTrain(featset):

x_ins = np.array([-1,-1,-1,-1])
y_ins = np.array([-1,-1,-1,-1])

for n in range(4):
x_ins[n] = in_x(featset[n,0], 10)
y_ins[n] = in_y(featset[n,1], 10)

# x_ins = x_ins.reshape(4,1)
# y_ins = y_ins.reshape(4,1)

inputs = np.array([x_ins, y_ins])


inputs = inputs.T

return inputs

# train network
def train_network(inputs, labels):
weights = np.random.rand(2,1)
weights /= 2.55
print(inputs)

for epochs in range(100):


print(weights)

# output value
Vz_hat = []
Iz_hat = np.dot(inputs, weights)
print(Iz_hat)
for I in Iz_hat:
Vz_hat.append(out_z(I, 10))
print(Vz_hat)

for num in range(4):


weights[0] += d_w1(inputs[num,0], Vz_hat[num])
weights[1] += d_w2(inputs[num,1], Vz_hat[num])
if weights[0] > .6125 or weights[1] > .6125:
break

if weights[0] > .6125 or weights[1] > .6125:


print(weights)
break

return weights

test_inputs = readSpikeTrain(feature_set)
fin_weights = train_network(test_inputs, labels1)

# final test for all cases


single_point1 = np.array([66.5, 67])
single_point2 = np.array([66.5, 66.5])
single_point3 = np.array([67, 66.5])
single_point4 = np.array([67, 67])

result1 = out_z(np.dot(single_point1, fin_weights), 10)


result2 = out_z(np.dot(single_point2, fin_weights), 10)
result3 = out_z(np.dot(single_point3, fin_weights), 10)
result4 = out_z(np.dot(single_point4, fin_weights), 10)

print("Final Weights:")
print(fin_weights)

print("Final Test Cases:")


print(single_point1)
print(result1)
print(single_point2)
print(result2)
print(single_point3)
print(result3)
print(single_point4)
print(result4)

You might also like