From the course: Applied AI for Human Resources

Unlock the full course today

Join today to access over 24,900 courses taught by industry experts.

Building a Keras rating model

Building a Keras rating model - Python Tutorial

From the course: Applied AI for Human Resources

Building a Keras rating model

- [Instructor] We will now build a Keras rating model based on the embeddings we built so far. Our recommendation strategy for this example is as follows. We build a model that can predict the rating, given an employee and a course. Then when we need to recommend training for a given employee, we predict ratings that this employee may give to each of the courses he or she has not taken. Then we recommend the courses with the highest predicted ratings. First, we split the ratings data into training and test datasets with a 10% hold-out for testing. We now, we'll continue to build the model. We create a fully connected Dense Layer with 128 units and an activation of relu. We pass the merged vector we prepared in the previous video as input. We create another fully connected layer with 32 units. Finally, we add a Dense Layer to predict the rating. We want to look at the absolute rating number and not a category. So we…

Contents