Skip to main content

How to create Predicted vs. Actual plot using abline_plot and statsmodels

I am trying to recreate this plot from this website in Python instead of R: enter image description here

Background

I have a dataframe called boston (the popular educational boston housing dataset).

I created a multiple linear regression model with some variables with statsmodels api below. Everything works.

import statsmodels.formula.api as smf
results = smf.ols('medv ~ col1 + col2 + ...', data=boston).fit()

I create a dataframe of actual values from the boston dataset and predicted values from above linear regression model.

new_df = pd.concat([boston['medv'], results.fittedvalues], axis=1, keys=['actual', 'predicted'])

This is where I get stuck. When I try to plot the regression line on top of the scatterplot, I get this error below.

from statsmodels.graphics.regressionplots import abline_plot
# scatter-plot data
ax = new_df.plot(x='actual', y='predicted', kind='scatter')

# plot regression line
abline_plot(model_results=results, ax=ax)

ValueError                                Traceback (most recent call last)
<ipython-input-156-ebb218ba87be> in <module>
      5 
      6 # plot regression line
----> 7 abline_plot(model_results=results, ax=ax)

/usr/local/lib/python3.8/dist-packages/statsmodels/graphics/regressionplots.py in abline_plot(intercept, slope, horiz, vert, model_results, ax, **kwargs)
    797 
    798     if model_results:
--> 799         intercept, slope = model_results.params
    800         if x is None:
    801             x = [model_results.model.exog[:, 1].min(),

ValueError: too many values to unpack (expected 2)

Here are the independent variables I used in the linear regression if that helps:

{'crim': {0: 0.00632, 1: 0.02731, 2: 0.02729, 3: 0.03237, 4: 0.06905},
 'chas': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'nox': {0: 0.538, 1: 0.469, 2: 0.469, 3: 0.458, 4: 0.458},
 'rm': {0: 6.575, 1: 6.421, 2: 7.185, 3: 6.998, 4: 7.147},
 'dis': {0: 4.09, 1: 4.9671, 2: 4.9671, 3: 6.0622, 4: 6.0622},
 'tax': {0: 296, 1: 242, 2: 242, 3: 222, 4: 222},
 'ptratio': {0: 15.3, 1: 17.8, 2: 17.8, 3: 18.7, 4: 18.7},
 'lstat': {0: 4.98, 1: 9.14, 2: 4.03, 3: 2.94, 4: 5.33},
 'rad3': {0: 0, 1: 0, 2: 0, 3: 1, 4: 1},
 'rad4': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'rad5': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'rad6': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'rad7': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'rad8': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'rad24': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0},
 'dis_sq': {0: 16.728099999999998,
  1: 24.67208241,
  2: 24.67208241,
  3: 36.75026884,
  4: 36.75026884},
 'lstat_sq': {0: 24.800400000000003,
  1: 83.53960000000001,
  2: 16.240900000000003,
  3: 8.6436,
  4: 28.4089},
 'nox_sq': {0: 0.28944400000000003,
  1: 0.21996099999999996,
  2: 0.21996099999999996,
  3: 0.209764,
  4: 0.209764},
 'rad24_lstat': {0: 0.0, 1: 0.0, 2: 0.0, 3: 0.0, 4: 0.0},
 'rm_lstat': {0: 32.743500000000004,
  1: 58.687940000000005,
  2: 28.95555,
  3: 20.57412,
  4: 38.09351},
 'rm_rad24': {0: 0.0, 1: 0.0, 2: 0.0, 3: 0.0, 4: 0.0}}


source https://stackoverflow.com/questions/74967916/how-to-create-predicted-vs-actual-plot-using-abline-plot-and-statsmodels

Comments

Popular posts from this blog

How to split a rinex file if I need 24 hours data

Trying to divide rinex file using the command gfzrnx but getting this error. While doing that getting this error msg 'gfzrnx' is not recognized as an internal or external command Trying to split rinex file using the command gfzrnx. also install'gfzrnx'. my doubt is I need to run this program in 'gfzrnx' or in 'cmdprompt'. I am expecting a rinex file with 24 hrs or 1 day data.I Have 48 hrs data in RINEX format. Please help me to solve this issue. source https://stackoverflow.com/questions/75385367/how-to-split-a-rinex-file-if-i-need-24-hours-data

ValueError: X has 10 features, but LinearRegression is expecting 1 features as input

So, I am trying to predict the model but its throwing error like it has 10 features but it expacts only 1. So I am confused can anyone help me with it? more importantly its not working for me when my friend runs it. It works perfectly fine dose anyone know the reason about it? cv = KFold(n_splits = 10) all_loss = [] for i in range(9): # 1st for loop over polynomial orders poly_order = i X_train = make_polynomial(x, poly_order) loss_at_order = [] # initiate a set to collect loss for CV for train_index, test_index in cv.split(X_train): print('TRAIN:', train_index, 'TEST:', test_index) X_train_cv, X_test_cv = X_train[train_index], X_test[test_index] t_train_cv, t_test_cv = t[train_index], t[test_index] reg.fit(X_train_cv, t_train_cv) loss_at_order.append(np.mean((t_test_cv - reg.predict(X_test_cv))**2)) # collect loss at fold all_loss.append(np.mean(loss_at_order)) # collect loss at order plt.plot(np.log(al...

Sorting large arrays of big numeric stings

I was solving bigSorting() problem from hackerrank: Consider an array of numeric strings where each string is a positive number with anywhere from to digits. Sort the array's elements in non-decreasing, or ascending order of their integer values and return the sorted array. I know it works as follows: def bigSorting(unsorted): return sorted(unsorted, key=int) But I didnt guess this approach earlier. Initially I tried below: def bigSorting(unsorted): int_unsorted = [int(i) for i in unsorted] int_sorted = sorted(int_unsorted) return [str(i) for i in int_sorted] However, for some of the test cases, it was showing time limit exceeded. Why is it so? PS: I dont know exactly what those test cases were as hacker rank does not reveal all test cases. source https://stackoverflow.com/questions/73007397/sorting-large-arrays-of-big-numeric-stings