diff --git a/README.md b/README.md index 3314be1..cedf35c 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,9 @@ # DATA301_projects -created on 2022-11-20 +## Contents +The main notebook is _project_1_eda_and_model_training.ipynb_. The notebook imports data from _diamonds.csv_, explores the data and sets up three regression models to predict diamond prices. + +The second notebook, _project_1_predictions.ipynb_, loads three trained models from the current directory and makes predictions based on a single row of data. + +The models are evaluated by comparing root mean square scores.