MulaiMulai sekarang secara gratis

Fit dan prediksi untuk regresi

Sekarang Anda telah melihat cara kerja regresi linear, tugas Anda adalah membuat model regresi linear berganda menggunakan semua fitur dalam himpunan data sales_df, yang telah dimuatkan untuk Anda. Sebagai pengingat, berikut dua baris pertama:

     tv        radio      social_media    sales
1    13000.0   9237.76    2409.57         46677.90
2    41000.0   15886.45   2913.41         150177.83

Anda kemudian akan menggunakan model ini untuk memprediksi penjualan berdasarkan nilai fitur uji.

LinearRegression dan train_test_split telah dimuatkan dari modul masing-masing.

Latihan ini adalah bagian dari kursus

Supervised Learning dengan scikit-learn

Lihat Kursus

Petunjuk latihan

  • Buat X, sebuah array yang berisi nilai semua fitur di sales_df, dan y, yang berisi semua nilai dari kolom "sales".
  • Instansiasikan model regresi linear.
  • Fit model ke data pelatihan.
  • Buat y_pred, yaitu prediksi untuk sales menggunakan fitur uji.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Create X and y arrays
X = sales_df.____("____", axis=____).____
y = sales_df["____"].____

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# Instantiate the model
reg = ____

# Fit the model to the data
____

# Make predictions
y_pred = reg.____(____)
print("Predictions: {}, Actual Values: {}".format(y_pred[:2], y_test[:2]))
Edit dan Jalankan Kode