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 tasks.

  • 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 dynamic().


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:

  • Price

  • House Size

  • Number of Bedrooms

  • Year Built

  • Number of Bathrooms

  • Number of Garage Spaces

  • Lot Size


  • An in-depth dive into dynamic workflows

  • How the Flyte type-system works

Code Walkthrough#

Gallery generated by Sphinx-Gallery