DEV Community

Cover image for Bring Machine Learning to Your Web Apps with TensorFlow.js
Abhay Singh Rathore
Abhay Singh Rathore

Posted on

Bring Machine Learning to Your Web Apps with TensorFlow.js

How to Implement Machine Learning in Your Web App using TensorFlow.js

Machine learning (ML) is no longer a lofty, out-of-reach concept. With libraries like TensorFlow.js, developers can now incorporate ML into their web applications. For example, you could create a system that recommends social media ads based on a user's views and searches.

This post is your guide to implementing ML using TensorFlow.js. We will discuss what TensorFlow.js is, how to use it, and how to implement a simple recommendation system in your web application.

Introduction to TensorFlow.js

TensorFlow.js is a JavaScript library developed by Google for training and deploying ML models in the browser and on Node.js. It allows you to develop ML models in JavaScript and use ML directly in the browser or in Node.js.

With TensorFlow.js, you can create new ML models from scratch, or you can use pre-trained models. Its flexibility and accessibility make it a popular choice among developers.

Setting up TensorFlow.js

To start using TensorFlow.js in your web app, you need to add the following script tag to your HTML file:

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@3.10.0/dist/tf.min.js"></script>
Enter fullscreen mode Exit fullscreen mode

Building a Recommendation System

Now, let's build a simple recommendation system that suggests social media ads based on user behaviour.

Step 1: Define your data

First, we need data for training. For this example, let's consider a simplified scenario where we only look at user's past ad clicks and the categories of those ads.

Our input data (features) will be the categories of ads that a user has clicked on in the past. Our output data (labels) will be the categories of ads that the user clicked on next.

In a real-world scenario, you would likely have much more data from various sources, such as user demographics, browsing history, etc.

Step 2: Preprocess your data

Before we feed our data to the model, we need to preprocess it. TensorFlow.js provides utilities for this. In our case, we'll encode our categorical data into numerical data that our model can understand.

Step 3: Define and train the model

Next, we'll define our model. We'll use a sequential model, which is a stack of layers where each layer has exactly one input tensor and one output tensor.

const model = tf.sequential();

model.add(tf.layers.dense({units: 10, inputShape: [numOfCategories]}));
model.add(tf.layers.dense({units: numOfCategories, activation: 'softmax'}));

model.compile({optimizer: 'sgd', loss: 'categoricalCrossentropy', metrics: ['accuracy']});
Enter fullscreen mode Exit fullscreen mode

Here, we have two layers. The first is our hidden layer, and the second is our output layer. The 'softmax' activation function ensures our output is a probability distribution over the ad categories.

Next, we train our model using our preprocessed data.

await model.fit(trainFeaturesTensor, trainLabelsTensor, {epochs: 100});
Enter fullscreen mode Exit fullscreen mode

Step 4: Make predictions

Once our model is trained, we can use it to make predictions. Here's how we'd predict the next ad category for a user:

const prediction = model.predict(userFeaturesTensor);
Enter fullscreen mode Exit fullscreen mode

This will give us a probability distribution over the ad categories. We can then recommend the ad with the highest probability.

Real-World Application: Social Media Ad Recommendations

Let's tie this back to our social media ad recommendation scenario.

Say, a user frequently views and clicks on ads related to technology and gadgets. Over time, our model will learn this pattern. When the

user logs in, our model would recommend ads from these categories with higher probability.

With TensorFlow.js, all of this happens right in the user's browser, which makes it faster and more efficient.

In summary, TensorFlow.js provides an accessible and powerful way to incorporate machine learning into your web applications. As we've seen, even with a few lines of JavaScript, we can start making personalized ad recommendations. Happy coding!


Thank you for reading this blog post! If you found it helpful and would like to stay updated on my latest articles and insights, I invite you to connect with me on LinkedIn and follow me on Twitter. Don't forget to subscribe to my newsletter, where I share exclusive content and updates directly to your inbox.

If you're interested in working together, discussing a project, or have any questions, feel free to reach out to me through direct message on LinkedIn or Twitter. I'm always happy to connect, collaborate, and help in any way I can.

Looking forward to staying in touch, and happy reading!

Top comments (0)