This code trains a linear regression model to predict the quality of red wine based on various features, and logs the resulting model's performance and parameters using MLflow. The code reads a CSV file from a URL, splits it into training and test sets, applies feature scaling to the training set, trains the model on the training set, and evaluates its performance on the test set using root mean squared error, mean absolute error, and R-squared metrics. The resulting model is then logged using MLflow, and can be stored in a model registry.
-
Notifications
You must be signed in to change notification settings - Fork 0
irenevanblerck/mlflow-test-example
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published