From the course: Applied AI for Human Resources
Join today to access over 24,900 courses taught by industry experts.
Predicting attrition with Keras - Python Tutorial
From the course: Applied AI for Human Resources
Predicting attrition with Keras
- [Instructor] In this video, we will use the model built in the earlier video to predict attrition. In order to predict attrition, we need to get the same set of six feature variables about the new employee for whom we need to predict attrition. Prediction then is an easy call to the model.predict_classes method that will output a zero or one based on the features. Let's run this code and review the prediction. We can see that the algorithm predicts that the employee will leave. We can also do bulk predictions for multiple employees. We need to create an array of arrays with the inner array representing an employee. The method call is the same. Let's run the code and review the results. I would strongly recommend using additional data points as well as experimenting with hyperparameters and layers for your own model. In the next chapter, we will solve a second use case discovering virtual teams.
Practice while you learn with exercise files
Download the files the instructor uses to teach the course. Follow along and learn by watching, listening and practicing.
Download courses and learn on the go
Watch courses on your mobile device without an internet connection. Download courses using your iOS or Android LinkedIn Learning app.