Predicting Magnetization Directions Using Convolutional Neural Networks