Issue 2.2: Model Training

[Ushna-Nadeem]

• [x] Develop and train a machine learning model using scikit-learn or tensorflow.
• [x] Perform hyperparameter tuning and optimization.