Build your first Machine Learning & Tensorflow model in 5 minutes! by@nattykohn

Build your first Machine Learning & Tensorflow model in 5 minutes!

June 23rd 2022 10,041 reads
Read on Terminal Reader
react to story with heart
react to story with light
react to story with boat
react to story with money
Nathaniel Kohn HackerNoon profile picture

Nathaniel Kohn

So you hear about Tensorflow, Machine Learning & Deep learning so much and you want to join the party :)
In this post we are going to build a model that will calculate:
Y = W*x + b
Hence when providing x to the model, it should predict the best possible Y.
We are going to train the model to give us the best W and b possible according to the what we’d train it on.
For example if we want our model to learn that W should be -1 and b 9, meaning Y = -1*x + 9; we should train it on many training pairs like this:
(x = input, Y = desired_output) = (x, Y) = (5, 4);
Y = -1*5 + 9 => Y = 4 (when the input is 5 predict Y to be 4)
(x, Y) = (3, 6); Y = -1*3 +9 => Y = 6 (when the input x is 3 predict Y to be 6)
The idea here is that we want the model to learn that W should be -1 and 9 and it would learn it with experience — the more training pairs (x, Y) we’d feed it the better it would learn to predict correctly.
So Let’s get down to business!
First we should start by setting up the environment.
Let’s first create a directory, run:
mkdir tf_linear_regression
Now (recommended) let’s run all in a virtual environment, follow these steps (it works for Mac, if you have a Linux or Windows try using Anaconda)
Install Python3
brew install python3
(If you don’t have Brew — just install it from here)
Now install the virtual environment
pip3 install virtualenv
Create the environment
cd tf_linear_regression
virtualenv -p python3 <path-to-our-dir>
Now activate it:
source <path-to-our-dir>/bin/activate
get out of the virtual env run:
To get out of the virtual environment run:

And now for the fun stuff

Install Tensorflow
pip3 install tensorflow
Now (just for fun) install Jyputer notebook (you don’t have to, you can copy the code below and paste it in a python file and run it as a Python script
pip3 install jupyter
Now let’s try to do something simple & have a look on the equation:
y = W*x + B
Assuming we have pairs of inputs and their labels => (X, Y) = [(x1,y1), (x2,y2), (x3,y3),.., (xn,yn)];
i.e for every input x we have the output y. This is a supervised machine learning (ML) where for every input we know the desired output.
Now we are going to train W and B in the equation above. Let assume we have (X, Y) = [ (5, 4), (3, 6), (-1, 10)…]
If we are going to guess what W and B would be…. (remember):
y = W*x + B
4 = []*5 +[]; 6 = []*3 + []; 10 = []*(-1) + []. you can see that W = -1 and
B = 9 would be a good fit us…
Now let’s build a model in Tensorflow (TF) which we’ll train to find those values for W and B. the training process in each step would be:
step1: for the input 5 the result is 4 , for the input 3 the result is 6 etc.
You can see that this is the same way as we are teaching a machine to recognize a cat in a picture: 1. “this input picture is a cat”; 2. “this input picture is a cat”; 3. “this input picture is not a cat” etc.
Let’s START!
Importing TF and Numpy:
Declaring x, y, w, b; Variables are used as trainables and Placeholders are for given train set inputs:
Declaring the training sets for X and for Y model input, and declaring the linear equation:
Now we’re setting the error function (model_delta), the loss function & the optimizer + initializing the variables we declared above:
Now this is a little counter intuitive, but in TF in contrary to program in Python, up until now nothing is running, now we should declare a session and then start the training process:
You can play with the number of iterations, you can see that for 100 we are not even close, but for 1000 it’s much better and 10000 we are nearly there:
  • 0.9999 ~ -1
  • 8.9998 ~ 9

react to story with heart
react to story with light
react to story with boat
react to story with money
. . . comments & more!