def backward(self, dz): dx, dw, db = layers.linear_backward(dz, self.cache1) return dx, dw, db # iris = datasets.load_iris() # X = iris.data # Y = iris.target # Y = to_categorical(iris.target, 3)