House Price Regression#
House Price Regression refers to the prediction of house prices based on various factors, using the XGBoost Regression model (in our case). In this example, we will train our data on the XGBoost model to predict house prices in multiple regions.
Where Does Flyte Fit In?#
Orchestrates the machine learning pipeline.
Helps cache the output state between
Easier backtracking to the error source.
Provides a Rich UI to view and manage the pipeline.
House price prediction pipeline for one region doesn’t require a
dynamic() workflow. When multiple regions are involved, to iterate through the regions at run-time and thereby build the DAG, Flyte workflow has to be
Refer to Dynamic Workflows section to learn more about dynamic workflows.
We will create a custom dataset to build our model by referring to the Sagemaker example.
The dataset will have the following columns:
Number of Bedrooms
Number of Bathrooms
Number of Garage Spaces
An in-depth dive into dynamic workflows
How the Flyte type-system works