当前位置: 首页 > 文档资料 > 机器学习 >

Linear Regression using Graphlab

优质
小牛编辑
134浏览
2023-12-01

Fire up graphlab create

import graphlab

Load some house value vs. crime rate data

Dataset is from Philadelphia, PA and includes average house sales price in a number of neighborhoods. The attributes of each neighborhood we have include the crime rate (‘CrimeRate’), miles from Center City (‘MilesPhila’), town name (‘Name’), and county name (‘County’).

sales = graphlab.SFrame.read_csv('Philadelphia_Crime_Rate_noNA.csv/')
Finished parsing file /Users/feisky/machine-learning/code/basic-classifiers/Philadelphia_Crime_Rate_noNA.csv
Parsing completed. Parsed 99 lines in 0.040707 secs.
Finished parsing file /Users/feisky/machine-learning/code/basic-classifiers/Philadelphia_Crime_Rate_noNA.csv
Parsing completed. Parsed 99 lines in 0.01334 secs.
------------------------------------------------------
Inferred types from first 100 line(s) of file as 
column_type_hints=[int,float,float,float,float,str,str]
If parsing fails due to incorrect types, you can correct
the inferred type list above and pass it to read_csv in
the column_type_hints argument
------------------------------------------------------
sales
HousePriceHsPrc ($10,000)CrimeRateMilesPhilaPopChgNameCounty
14046314.046329.710.0-1.0AbingtonMontgome
11303311.303324.118.04.0AmblerMontgome
12418612.418619.525.08.0AstonDelaware
11049011.04949.425.02.7BensalemBucks
791247.912454.119.03.9Bristol B.Bucks
926349.263448.620.00.6Bristol T.Bucks
892468.924630.815.0-2.6BrookhavenDelaware
19514519.514510.820.0-3.5Bryn AthynMontgome
29734229.734220.214.00.6Bryn MawrMontgome
26429826.429820.426.06.0BuckinghamBucks

[99 rows x 7 columns]
Note: Only the head of the SFrame is printed.
You can use print_rows(num_rows=m, num_columns=n) to print more rows and columns.

Exploring the data

The house price in a town is correlated with the crime rate of that town. Low crime towns tend to be associated with higher house prices and vice versa.

graphlab.canvas.set_target('ipynb')
sales.show(view="Scatter Plot", x="CrimeRate", y="HousePrice")

Fit the regression model using crime as the feature

crime_model = graphlab.linear_regression.create(sales, target='HousePrice', features=['CrimeRate'],validation_set=None,verbose=False)

Let’s see what our fit looks like

Matplotlib is a Python plotting library that is also useful for plotting. You can install it with:

‘pip install matplotlib’

import matplotlib.pyplot as plt
%matplotlib inline
plt.plot(sales['CrimeRate'],sales['HousePrice'],'.',
        sales['CrimeRate'],crime_model.predict(sales),'-')
[<matplotlib.lines.Line2D at 0x118f2d150>,
 <matplotlib.lines.Line2D at 0x118f2d390>]

Above: blue dots are original data, green line is the fit from the simple regression.

Remove Center City and redo the analysis

Center City is the one observation with an extremely high crime rate, yet house prices are not very low. This point does not follow the trend of the rest of the data very well. A question is how much including Center City is influencing our fit on the other datapoints. Let’s remove this datapoint and see what happens.

sales_noCC = sales[sales['MilesPhila'] != 0.0] 
sales_noCC.show(view="Scatter Plot", x="CrimeRate", y="HousePrice")

Refit our simple regression model on this modified dataset:

crime_model_noCC = graphlab.linear_regression.create(sales_noCC, target='HousePrice', features=['CrimeRate'],validation_set=None, verbose=False)

Look at the fit:

plt.plot(sales_noCC['CrimeRate'],sales_noCC['HousePrice'],'.',
        sales_noCC['CrimeRate'],crime_model.predict(sales_noCC),'-')
[<matplotlib.lines.Line2D at 0x1172d8050>,
 <matplotlib.lines.Line2D at 0x1172d8290>]

Compare coefficients for full-data fit versus no-Center-City fit

Visually, the fit seems different, but let’s quantify this by examining the estimated coefficients of our original fit and that of the modified dataset with Center City removed.

crime_model.get('coefficients')
nameindexvaluestderr
(intercept)None176626.04688111245.5882194
CrimeRateNone-576.804949058226.90225951

[2 rows x 4 columns]

crime_model_noCC.get('coefficients')
nameindexvaluestderr
(intercept)None225204.60430316404.0247514
CrimeRateNone-2287.69717443491.537478123

[2 rows x 4 columns]

Above: We see that for the “no Center City” version, per unit increase in crime, the predicted decrease in house prices is 2,287. In contrast, for the original dataset, the drop is only 576 per unit increase in crime. This is significantly different!

High leverage points:

Center City is said to be a “high leverage” point because it is at an extreme x value where there are not other observations. As a result, recalling the closed-form solution for simple regression, this point has the potential to dramatically change the least squares line since the center of x mass is heavily influenced by this one point and the least squares line will try to fit close to that outlying (in x) point. If a high leverage point follows the trend of the other data, this might not have much effect. On the other hand, if this point somehow differs, it can be strongly influential in the resulting fit.

Influential observations:

An influential observation is one where the removal of the point significantly changes the fit. As discussed above, high leverage points are good candidates for being influential observations, but need not be. Other observations that are not leverage points can also be influential observations (e.g., strongly outlying in y even if x is a typical value).

Remove high-value outlier neighborhoods and redo analysis

Based on the discussion above, a question is whether the outlying high-value towns are strongly influencing the fit. Let’s remove them and see what happens.

sales_nohighend = sales_noCC[sales_noCC['HousePrice'] < 350000] 
crime_model_nohighend = graphlab.linear_regression.create(sales_nohighend, target='HousePrice', features=['CrimeRate'],validation_set=None, verbose=False)

Do the coefficients change much?

crime_model_noCC.get('coefficients')
nameindexvaluestderr
(intercept)None225204.60430316404.0247514
CrimeRateNone-2287.69717443491.537478123

[2 rows x 4 columns]

crime_model_nohighend.get('coefficients')
nameindexvaluestderr
(intercept)None199073.58961511932.5101105
CrimeRateNone-1837.71280989351.519609333

[2 rows x 4 columns]

Above: We see that removing the outlying high-value neighborhoods has some effect on the fit, but not nearly as much as our high-leverage Center City datapoint.