This site runs best with JavaScript enabled.Summer of AI

Lesson 4 Code

Combining Numerical and Categorical Data

In this lesson we'll look back at the house price dataset that we used in lesson 2. Previously, we just used 4 numerical columns for price prediction, but now we'll add 2 categorical columns as well. We'll use embedding layers to process those categories, and then see if we can get a better accuracy than before.

House dataset:

Lesson 2 House Price Code:

Completed Lesson 4 Code: