https://mingredbestcheapprice.blogspot.com/2022/08/sklearn-prediction-on-test-dataset-with.html