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
Petunjuk latihan
- Buat
X, sebuah array yang berisi nilai semua fitur disales_df, dany, yang berisi semua nilai dari kolom"sales". - Instansiasikan model regresi linear.
- Fit model ke data pelatihan.
- Buat
y_pred, yaitu prediksi untuksalesmenggunakan 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]))