import keras import numpy as np A, b = np.load("A.npy"), np.load("b.npy") W_0, x_0 = np.load("W_0.npy"), np.load("x_0.npy")