Embedding-based neural network for investment return prediction