I have a project on iris classifier, I have done most of the steps but I am stuck. How do I:

1 Update the Python code in train.py so that it can:

Save the confusion matrix as a PNG image in an outputs/ folder.

Save the trained model in the same outputs/ folder (e.g., using the joblib library).