Memvisualisasikan model regresi linear
Sekarang setelah Anda membangun model regresi linear dan melatihnya menggunakan semua observasi yang tersedia, Anda dapat memvisualisasikan seberapa baik model memodelkan data. Ini memungkinkan Anda menafsirkan hubungan antara pengeluaran iklan radio dan nilai sales.
Variabel X, sebuah array berisi nilai radio, y, sebuah array berisi nilai sales, serta predictions, sebuah array berisi nilai prediksi model untuk y berdasarkan X, semuanya telah dimuat untuk Anda dari latihan sebelumnya.
Latihan ini adalah bagian dari kursus
Supervised Learning dengan scikit-learn
Petunjuk latihan
- Impor
matplotlib.pyplotsebagaiplt. - Buat plot sebar yang memvisualisasikan
yterhadapX, dengan observasi berwarna biru. - Gambar plot garis berwarna merah yang menampilkan prediksi terhadap
X. - Tampilkan plot tersebut.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import matplotlib.pyplot
import ____.____ as ____
# Create scatter plot
plt.scatter(____, ____, color="____")
# Create line plot
plt.plot(____, ____, color="____")
plt.xlabel("Radio Expenditure ($)")
plt.ylabel("Sales ($)")
# Display the plot
plt.____()