Как сохранить прогнозируемые результаты в нейронной сети в матрицу в Python2.7

Question

У меня есть следующий код

for i in xrange(100):

print "Input: \n" + str(X) 

print "Actual Output: \n" + str(y) 

print "Predicted Output: \n" + str(NN.forward(X)) 

print "Loss: \n" + str(np.mean(np.square(y - NN.forward(X)))) # mean sum squared loss

print "\n"

NN.train(X, y)

После запуска этого кода я получаю прогнозируемый вывод в консоли, но я хочу сохранить его как переменную (матрицу).

Как сохранить

#print "Predicted Output: \n" + str(NN.forward(X)) #

значения в матричной форме.

Спасибо.