In this recipe, we created a regression neural network with three fully-connected hidden layers to predict the birth weight of the low birth weight data set. When comparing this to a logistic output to predict above or below 2,500 grams, we achieved similar results and achieved them in fewer generations. In the next recipe, we will try to improve our logistic regression by making it a multiple-layer, logistic-type neural network.
How it works...
Get TensorFlow Machine Learning Cookbook - Second Edition now with the O’Reilly learning platform.
O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.
