Hello there,
I am a freelance engineer with good experience in developing and training ML models in Python. I can provide a quick solution for your issue. Please message me and share more details, if you would like to hire me.
update: I saw ur code, but need the data to debug the issue. If you can send me the data, I will be happy to spend some time debugging it. If I am able to give a good solution, then only you can award me the project.
Regards,
Amit R