Skip to content

Instantly share code, notes, and snippets.

@mempirate

mempirate/test_viz.py

Created April 4, 2018 15:21
Show Gist options
  • Save mempirate/d6144baa99d8fc0a646259655ac831c8 to your computer and use it in GitHub Desktop.
Save mempirate/d6144baa99d8fc0a646259655ac831c8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
test_viz.py
# Visualizing the test set results
plt.figure(figsize = (12, 6))
plt.scatter(X_test, y_test, alpha = 0.4, label = 'Observation Points')
plt.plot(X_train, regressor.predict(X_train), color = 'green', lw = 0.7, alpha = 0.8, label = 'Best Fit Line')
plt.title('Linear Regression (Test Set)')
plt.xlabel('Years of Experience')
plt.ylabel('Salary')
plt.legend()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment