Has anyone fixed the TODOs in the train_model function regarding the class weight value calculation and the best weights?