Geological information prediction for shield machine using an enhanced multi-head self-attention convolution neural network with two-stage feature extraction