Dear Osuman: I think that you should post your code to stackoverflow and get helps from there. A vector in shape (1,4) seems to be a prediction of one image over 4 classes, not a feature of the image.