This project classifies the Iris dataset using Sklearn Logistic Regression, XGBoost, SVM, and PyTorch models with hyperparameter optimization using Optuna. Model training, evaluation, and storage are done using MLflow.
- Install dependencies:
- Run the main script:
- Clean up the project folder:
- Access the MLflow UI:Navigate to http://127.0.0.1:5000 in your web browser to access the MLflow experiments.
This project is licensed under the MIT License.