Skip to content## Building a Simple Neural Network | TensorFlow for Hackers (Part II)

# 1. Preparing the data

# 2. Exploration

# 3. Building our model

## 3.1 Encoding the data

## 3.2 Splitting the data

## 3.3 Building our Neural Network

## 3.4 Training

# 4. Evaluation

# 5. Conclusion(s)

# References

## Want to be a Machine Learning expert?

— Deep Learning, Neural Networks, TensorFlow, Python — 4 min read

Share

Part 2 deserves its own introductory video:

In this one, you will learn how to create a Neural Network (NN) and use it for deciding whether a student has alcohol consumption problems.

Do students drink too much? How can you predict that? What predicts it best? How much too much is exactly?

Those questions might be difficult to answer, yet we can start somewhere. We can use a very limited dataset to get a sense of what the answers might look like. Something like this one.

The dataset contains 1044 instances and 32 variables (most of which binary and categorical). Actually, it consists of 2 other datasets. The first provides data for students enrolled in Portuguese class. The second describes students enrolled in a math course. There is overlap (yep, I know) between the datasets, that is some students attend both classes.

Let’s build an NN model for classifying whether a student has alcohol consumption problem. For that, we will use our trusty old friend - TensorFlow.

Before getting there, we have a bit of dirty work to do. Our dataset is not clean enough to just start and feed the data to our NN model. A bit of wrangling is required. But first, let’s start with some setting up:

`1import tensorflow as tf2import numpy as np3import matplotlib.pyplot as plt4import pandas as pd5import seaborn as sns6from math import floor, ceil7from pylab import rcParams89%matplotlib inline`

Some styling and making our experiments reproducible:

`1sns.set(style='ticks', palette='Spectral', font_scale=1.5)23material_palette = ["#4CAF50", "#2196F3", "#9E9E9E", "#FF9800", "#607D8B", "#9C27B0"]4sns.set_palette(material_palette)5rcParams['figure.figsize'] = 16, 867plt.xkcd();8random_state = 429np.random.seed(random_state)10tf.set_random_seed(random_state)`

Remember, our data is stored in two separate files. Let’s load them, assign proper course attendance to each student and merge them into one:

`1math_df = pd.read_csv("data/student/student-mat.csv", sep=";")2port_df = pd.read_csv("data/student/student-por.csv", sep=";")34math_df["course"] = "math"5port_df["course"] = "portuguese"67merged_df = math_df.append(port_df)8merged_df.shape`

`1(1044, 34)`

Exactly as promised - 1044 rows, but we have duplicates. The dataset archive contains instructions on how to find them. The merged result contains 382 instances. We will update the course column for those students, too:

`1merge_vector = ["school","sex","age","address",2 "famsize","Pstatus","Medu","Fedu",3 "Mjob","Fjob","reason","nursery","internet"]45duplicated_mask = merged_df.duplicated(keep=False, subset=merge_vector)6duplicated_df = merged_df[duplicated_mask]7unique_df = merged_df[~duplicated_mask]8both_courses_mask = duplicated_df.duplicated(subset=merge_vector)9both_courses_df = duplicated_df[~both_courses_mask].copy()10both_courses_df["course"] = "both"11students_df = unique_df.append(both_courses_df)`

We will use the following formula to quantify the amount of alcohol taken during the week per student:

$Alcohol=\frac{Walc \times 2 \,+\, Dalc \times 5}{7}$The new value changes in the interval $[1, 5]$. Furthermore, we will classify student as a drinker if that value is greater than 2.

`1students_df = students_df.sample(frac=1)2students_df['alcohol'] = (students_df.Walc * 2 + students_df.Dalc * 5) / 73students_df['alcohol'] = students_df.alcohol.map(lambda x: ceil(x))4students_df['drinker'] = students_df.alcohol.map(lambda x: "yes" if x > 2 else "no")`

Finally, we can get a feel for our data. Let’s take a look at the course distribution:

`1students_df.course.value_counts().plot(kind="bar", rot=0);`

And the alcohol consumption from the formula:

`1students_df.alcohol.value_counts().plot(kind="bar", rot=0);`

The actual variable that we are going to predict:

`1students_df.drinker.value_counts().plot(kind="bar", rot=0);`

Somewhat more comprehensive overview:

`1sns.pairplot(students_df[['age', 'absences', 'G3', 'goout', 'freetime', 'studytime', 'drinker']], hue='drinker');`

Let’s have a look at a general correlations matrix:

`1corr_mat = students_df.corr()2fig, ax = plt.subplots(figsize=(20, 12))3sns.heatmap(corr_mat, vmax=1.0, square=True, ax=ax);`

It is time for the fun part. Well, not just yet.

Most of our variables are categorical and we must one-hot encode them four our NN to work properly. First, let’s define a little helper function:

`1def encode(series):2 return pd.get_dummies(series.astype(str))`

Our features and target variable using our little helper function:

`1train_x = pd.get_dummies(students_df.school)2train_x['age'] = students_df.age3train_x['absences'] = students_df.absences4train_x['g1'] = students_df.G15train_x['g2'] = students_df.G26train_x['g3'] = students_df.G37train_x = pd.concat([train_x, encode(students_df.sex), encode(students_df.Pstatus),8 encode(students_df.Medu), encode(students_df.Fedu),9 encode(students_df.guardian), encode(students_df.studytime),10 encode(students_df.failures), encode(students_df.activities),11 encode(students_df.higher), encode(students_df.romantic),12 encode(students_df.reason), encode(students_df.paid),13 encode(students_df.goout), encode(students_df.health),14 encode(students_df.famsize), encode(students_df.course)15 ], axis=1)1617train_y = encode(students_df.drinker)`

Let’s allocate 90% of the data for training and use 10% for testing:

`1train_size = 0.923train_cnt = floor(train_x.shape[0] * train_size)4x_train = train_x.iloc[0:train_cnt].values5y_train = train_y.iloc[0:train_cnt].values6x_test = train_x.iloc[train_cnt:].values7y_test = train_y.iloc[train_cnt:].values`

Our NN consists of input, output and 1 hidden layer. We are using ReLU as activation function of the hidden layer and softmax for our output layer. As an additional bonus we will use Dropout - simple way to reduce overfitting during the training of our network. Let’s wrap our model in a little helper function:

`1def multilayer_perceptron(x, weights, biases, keep_prob):2 layer_1 = tf.add(tf.matmul(x, weights['h1']), biases['b1'])3 layer_1 = tf.nn.relu(layer_1)4 layer_1 = tf.nn.dropout(layer_1, keep_prob)5 out_layer = tf.matmul(layer_1, weights['out']) + biases['out']6 return out_layer`

Let’s set the number of neurons in the hidden layer to 38 and randomly initialize the weights and biases considering their proper dimensions:

`1n_hidden_1 = 382n_input = train_x.shape[1]3n_classes = train_y.shape[1]45weights = {6 'h1': tf.Variable(tf.random_normal([n_input, n_hidden_1])),7 'out': tf.Variable(tf.random_normal([n_hidden_1, n_classes]))8}910biases = {11 'b1': tf.Variable(tf.random_normal([n_hidden_1])),12 'out': tf.Variable(tf.random_normal([n_classes]))13}1415keep_prob = tf.placeholder("float")`

We will train our model for 5,000 epochs (training steps) with a batch size of 32. That is, at each step, we will train our NN using 32 rows of our data. Granted, in our case you can just train on the whole dataset. However, when the data is huge and you can’t fit it in memory, you would love to split it and feed it to the model at batches (chunks):

`1training_epochs = 50002display_step = 10003batch_size = 3245x = tf.placeholder("float", [None, n_input])6y = tf.placeholder("float", [None, n_classes])`

In order for our model to learn, we need to define what is good. Actually, we will define what is bad and try to minimize it. We will call the “badness” - error or cost (hence, the cost function). It represents how far off of the true result our model is at some point during training. We would love that error to be 0 for all possible inputs. Currently, that happens only in Sci-Fi novels (not that I discourage dreaming about it).

The cost function that we are going to use is called “Cross-Entropy”. It is defined as:

$H_{y'}(y) = -\sum_i y'_i \log(y_i)$Where $y$ is the predicted distribution for our alcohol consumption and $y'$ is the ground truth. This guide might be helpful for better understanding Cross-Entropy. TensorFlow has a little helper function with the sweet little name `softmax_cross_entropy_with_logits`

. It use softmax as activation function for our output layer and use Cross-Entropy as error function.

`1cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=predictions, labels=y))`

Now, for the actual workhorse - Adam (nope, not the from the Bible - although, that would’ve been fun). Adam is a type of gradient descent optimization algorithm which essentially tries as hard as he can to find proper weights and biases for our network via minimizing the cost function that we specified above. It is well beyond the scope of this post to describe Adam in details, but you can find all the necessary information over here - with tons of nice pictures!

Using Adam in TensorFlow is quite easy, we just have to specify learning rate (you can fiddle with that one) and pass the cost function we defined above:

`1optimizer = tf.train.AdamOptimizer(learning_rate=0.0001).minimize(cost)`

Our model is created by just calling our helper function with the proper arguments:

`1predictions = multilayer_perceptron(x, weights, biases, keep_prob)`

Our finished NN looks something like this (much reduced input and hidden layer sizes):

Time to see how well our model can predict. During the training, we will set the keep probability of the Dropout to 0.8 and reset it to 1.0 during test time:

`1with tf.Session() as sess:2 sess.run(tf.global_variables_initializer())34 for epoch in range(training_epochs):5 avg_cost = 0.06 total_batch = int(len(x_train) / batch_size)7 x_batches = np.array_split(x_train, total_batch)8 y_batches = np.array_split(y_train, total_batch)9 for i in range(total_batch):10 batch_x, batch_y = x_batches[i], y_batches[i]11 _, c = sess.run([optimizer, cost], feed_dict={x: batch_x,12 y: batch_y, keep_prob: 0.8})13 avg_cost += c / total_batch14 if epoch % display_step == 0:15 print("Epoch:", '%04d' % (epoch+1), "cost=", \16 "{:.9f}".format(avg_cost))1718 print("Optimization Finished!")1920 correct_prediction = tf.equal(tf.argmax(predictions, 1), tf.argmax(y, 1))21 accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))22 print("Accuracy:", accuracy.eval({x: x_test, y: y_test, keep_prob: 1.0}))`

`1Epoch: 0001 cost= 103.3465877112Epoch: 1001 cost= 2.0532956983Epoch: 2001 cost= 0.4641090084Epoch: 3001 cost= 0.3045922875Epoch: 4001 cost= 0.2841830746Optimization Finished!7Accuracy: 0.731343`

Yes, you did it! You survived another part of this tutorial. But what did you achieved? Our model got roughly 73% accuracy on the test set. Is this good? Well… no, it is not!

How is that possible? The authors of the paper linked from the dataset attained 92% accuracy. Which is (as they state) acceptable. So, why our model performs so badly?

For one thing, we excluded overlapping student data, which made our dataset considerably smaller - from 1044 to just 662 instances (I haven’t found any type of duplicate reduction technique used by the authors. Please, write me a comment if I am wrong about that one). Due to the high prevalence of no drinkers, this might have a decremental effect on our model performance.

Of course, you can try different parameters, architecture, training epochs etc… Feel free to do so! Till next time!

Student Alcohol Consumption - Description of the used dataset

Using Data Mining to Predict Secondary School Student Alcohol Consumption - A paper using this dataset and comparing 3 different models on it (including NN)

Student Alcohol Consumption Prediction - Possibly source code used in the previous paper

MNIST classification using TensorFlow - Use Deep Neural Network to classify handwritten digits

How to choose the number of hidden layers and neurons in NN?

How to handle ordinal data in NN models - Lots of the variables are ordinal. This paper presents an approach to handling that kind of data in NN models

Simpler way to handle ordinal data in NN models

Share

Join the weekly newsletter on Data Science, Deep Learning and Machine Learning in your inbox, curated by me! Chosen by **10,000+** Machine Learning practitioners. (There might be some exclusive content, too!)

You'll never get spam from me