Apache Spark 2.x Cookbook

Doing binary classification using SVM

Classification is a technique to put data into different classes based on its utility. For example, an e-commerce company can apply two labels, namely will buy or will not buy, to the potential visitors.

This classification is done by providing some already labeled data to machine-learning algorithms called training data, as you know already. The challenge is how to mark the boundary between the two classes. Let's take a simple example, as shown in the following figure:

In the preceding case, we designated gray and black to the "will not buy" and "will buy" labels, respectively. Here, drawing a line between ...