MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor LinearRegression.
  • Instansiasikan model regresi linear.
  • Prediksi nilai penjualan menggunakan X, simpan sebagai predictions.

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])
Edit dan Jalankan Kode