decision_tree.joblib | ||
diamonds.csv | ||
multiple_regression.joblib | ||
project_1_eda_and_model_training.ipynb | ||
project_1_predictions.ipynb | ||
README.md | ||
simple_regression.joblib |
DATA301_projects
Project 1 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.