Search icon CANCEL
Arrow left icon
Explore Products
Best Sellers
New Releases
Books
Videos
Audiobooks
Learning Hub
Conferences
Free Learning
Arrow right icon
Arrow up icon
GO TO TOP
Deep Learning with MXNet Cookbook

You're reading from   Deep Learning with MXNet Cookbook Discover an extensive collection of recipes for creating and implementing AI models on MXNet

Arrow left icon
Product type Paperback
Published in Dec 2023
Publisher Packt
ISBN-13 9781800569607
Length 370 pages
Edition 1st Edition
Languages
Tools
Arrow right icon
Author (1):
Arrow left icon
Andrés P. Torres Andrés P. Torres
Author Profile Icon Andrés P. Torres
Andrés P. Torres
Arrow right icon
View More author details
Toc

Table of Contents (12) Chapters Close

Preface 1. Chapter 1: Up and Running with MXNet FREE CHAPTER 2. Chapter 2: Working with MXNet and Visualizing Datasets – Gluon and DataLoader 3. Chapter 3: Solving Regression Problems 4. Chapter 4: Solving Classification Problems 5. Chapter 5: Analyzing Images with Computer Vision 6. Chapter 6: Understanding Text with Natural Language Processing 7. Chapter 7: Optimizing Models with Transfer Learning and Fine-Tuning 8. Chapter 8: Improving Training Performance with MXNet 9. Chapter 9: Improving Inference Performance with MXNet 10. Index 11. Other Books You May Enjoy

Understanding math for classification models

As we saw in the previous chapter, classification problems are supervised learning problems whose output is a class from a set of classes (categorical assignments) – for example, the iris class of a flower.

As we will see throughout this recipe, classification models can be seen as individual cases of regression models. We will start by exploring a binary classification model. This is a model that will output one of two classes. We will label these classes [0, 1] for simplicity.

The simplest model we can use for such a binary classification problem is a linear regression model. This model will output a number; therefore, to modify the output to satisfy our new classification criteria, we will modify the activation function to a more suitable one.

As in the previous recipes, we will use a neural network as our model, and we will solve the iris dataset prediction problem we introduced in the second recipe, Toy dataset for classification...

lock icon The rest of the chapter is locked
Register for a free Packt account to unlock a world of extra content!
A free Packt account unlocks extra newsletters, articles, discounted offers, and much more. Start advancing your knowledge today.
Unlock this book and the full library FREE for 7 days
Get unlimited access to 7000+ expert-authored eBooks and videos courses covering every tech area you can think of
Renews at €18.99/month. Cancel anytime