Mulai sekarangMulai 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 merupakan bagian dari kursus

Supervised Learning dengan scikit-learn

Lihat Kursus

Instruksi latihan

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

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# Import LinearRegression
from ____.____ import ____

# Create the model
reg = ____()

# Fit the model to the data
____

# Make predictions
predictions = ____

print(predictions[:5])
Edit dan Jalankan Kode