Improved sub-nuclear prediction by Deep Learning using an augmented dataset