def predict(x):
    net = load('c.pkl')
    result = [np.argmax(net.feedforward(x_a)) for x_a in x]
    result = np.reshape(result, newshape=(x.shape[0], 1))
    return result