Membangun model regresi linear
Sekarang setelah Anda membuat array fitur dan target, Anda akan melatih model regresi linear pada semua nilai fitur dan target.
Karena tujuannya adalah menilai hubungan antara nilai fitur dan target, tidak perlu membagi data menjadi himpunan latih dan uji.
X dan y telah dimuat sebelumnya untuk Anda sebagai berikut:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Latihan ini adalah bagian dari kursus
Supervised Learning dengan scikit-learn
Petunjuk latihan
- Impor
LinearRegression. - Instansiasikan model regresi linear.
- Prediksi nilai penjualan menggunakan
X, simpan sebagaipredictions.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])