Machine Learning With Go

Linear regression example

To illustrate linear regression, let's take an example problem and create our first machine learning model! The example data that we are going to use is example advertising data. It is in the .csv format and looks as follows:

$ head Advertising.csv 
TV,Radio,Newspaper,Sales
230.1,37.8,69.2,22.1
44.5,39.3,45.1,10.4
17.2,45.9,69.3,9.3
151.5,41.3,58.5,18.5
180.8,10.8,58.4,12.9
8.7,48.9,75,7.2
57.5,32.8,23.5,11.8
120.2,19.6,11.6,13.2
8.6,2.1,1,4.8

The dataset includes a set of attributes representing spend on advertising outlets (TV, Radio, and Newspaper) along with corresponding sales (Sales). Our goal in this example will be to model the sales (our dependent variable) by one of the attributes of advertising spend (our ...