def eaNetLinear(eaData, L=4, n=4):
net = Ea()
net.L = L
net.n[0] = eaData.X.shape[0] # nx
for l in range(1, L):
net.n[l] = n
net.activFun[l] = eaActivFun.linear
net.n[L] = 1
net.activFun[L] = eaActivFun.linear
net.costFun = eaCostFun.L2
return net
def eaNetLogistics(eaData):
net = Ea()
net.L = 2
net.n[0] = eaData.X.shape[0] # nx
net.n[1] = 20
net.n[2] = 4
net.n[3] = 4
net.n[net.L] = 1
net.activFun[1] = eaActivFun.relu
net.activFun[2] = eaActivFun.relu
net.activFun[3] = eaActivFun.sigmoid
net.activFun[net.L] = eaActivFun.sigmoid
net.costFun = eaCostFun.L3
return net