Machine Learning With Go

Choosing our independent variable

So, now we have some intuition about our data and have come to terms with how our data fits within the assumptions of the linear regression model. Now, how do we choose which variable to use as our independent variable in trying to predict our dependent variable, and average points per game?

The easiest way to make this decision is by visually exploring the correlation between the dependent variable and all of the choices that you have for independent variables. In particular, you can make scatter plots (using gonum.org/v1/plot) of your dependent variable versus each of the other variables:

// Open the advertising dataset file. f, err := os.Open("Advertising.csv") if err != nil { log.Fatal(err) } defer f.Close() ...