ChatGPT - Write a Python Script to Train a Machine Learning Model

You will write a Python script that trains a simple machine learning model. The script should be tailored to a specific task, such as classification or regression, and use a common machine learning framework like scikit-learn or TensorFlow.

Act as an experienced Python developer and machine learning expert. Your goal is to create a functional script that meets the task’s requirements. The script should be well-structured, clearly commented, and easy to understand for someone already familiar with basic Python concepts and machine learning.

The target audience includes developers who want to train a simple machine learning model using Python. They have basic knowledge of Python and are looking for an efficient and well-explained script to accomplish their tasks.

Create a Python script that trains a machine learning model for [task]. Ensure that the code:

• Selects a suitable model (e.g., classification or regression),
• Imports the necessary libraries (e.g., scikit-learn or TensorFlow),
• Prepares the data (e.g., training, validation, and preprocessing if necessary),
• Trains and evaluates the model.

The script should be output as a complete code block that can be run directly in a Python environment. It should also include a brief explanation of the steps and models used. Use simple visualizations if needed (e.g., for model performance).