Book Image

Machine Learning Solutions

Book Image

Machine Learning Solutions

Overview of this book

Machine learning (ML) helps you find hidden insights from your data without the need for explicit programming. This book is your key to solving any kind of ML problem you might come across in your job. You’ll encounter a set of simple to complex problems while building ML models, and you'll not only resolve these problems, but you’ll also learn how to build projects based on each problem, with a practical approach and easy-to-follow examples. The book includes a wide range of applications: from analytics and NLP, to computer vision domains. Some of the applications you will be working on include stock price prediction, a recommendation engine, building a chat-bot, a facial expression recognition system, and many more. The problem examples we cover include identifying the right algorithm for your dataset and use cases, creating and labeling datasets, getting enough clean data to carry out processing, identifying outliers, overftting datasets, hyperparameter tuning, and more. Here, you'll also learn to make more timely and accurate predictions. In addition, you'll deal with more advanced use cases, such as building a gaming bot, building an extractive summarization tool for medical documents, and you'll also tackle the problems faced while building an ML model. By the end of this book, you'll be able to fine-tune your models as per your needs to deliver maximum productivity.
Table of Contents (19 chapters)
Machine Learning Solutions
Foreword
Contributors
Preface
Index

Selecting the machine learning algorithm


We already know that we are using Convolution Neural Networks (CNN) for developing this application. You might wonder why we have chosen CNN and not another neural net. You might already know the answer to this question. There are three reasons why we have chosen CNN:

  • The amount of visual data present nowadays, which is carefully hand-labeled

  • The affordable computation machines through which GPUs open the door for optimization

  • The various kinds of architecture of CNN outperforms the other algorithms

Due to these reasons, we have chosen the CNN with SSD. During the development of the baseline model, we will be using MobileNet, which uses CNN with Single Shot Detector (SSD) techniques underneath. So, in this section, we will look at the architecture of the CNN used during the development of the MobileNet. This will help us understand the pre-trained model.

Architecture of the MobileNet SSD model

MobileNet SSD is fast and does the job of object detection...