Getting ready

The same maximum margin concept can be applied toward fitting linear regression. Instead of maximizing the margin that separates the classes, we can think about maximizing the margin that contains the most (x, y) points. To illustrate this, we will use the same iris dataset, and show that we can use this concept to fit a line between sepal length and petal width.

The corresponding loss function will be similar to . Here,is half of the width of the margin, which makes the loss equal to 0 if a point lies in this region.

Get TensorFlow Machine Learning Cookbook - Second Edition now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.