Advertisement
vlpap

getNNI

May 8th, 2024 (edited)
530
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.63 KB | None | 0 0
  1. def getNNI(W1, b1, W2, b2, alpha_s: list, beta_s: list, n):
  2.     a = np.asarray(alpha_s)
  3.     b = np.asarray(beta_s)
  4.     ba_term = np.prod(b - a)
  5.     ab = np.array([a, b])
  6.    
  7.     X = []
  8.     L = get_L(n)
  9.     for l in L:
  10.         x = - b1 - W1[0]*ab[l[0],0] - W1[1]*ab[l[1],1] - W1[2]*ab[l[2],2] - W1[3]*ab[l[3],3] - W1[4]*ab[l[4],4] - W1[5]*ab[l[5],5]
  11.         X.append(x)
  12.     Xsi = get_Xsi(n, False)
  13.    
  14.     Fi = np.array([xsi * Li(n, -np.exp(x, dtype=np.float128)) for xsi, x in zip(Xsi, X)])
  15.     Fi = np.sum(Fi, axis=0)
  16.    
  17.     I = b2[0] * ba_term + np.sum(W2.flatten() * (ba_term + Fi / np.prod(W1, axis=0) ))
  18.     return I
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement