Xây dựng mô hình hồi quy tuyến tính
Bây giờ khi bạn đã tạo xong mảng đặc trưng và mục tiêu, bạn sẽ huấn luyện một mô hình hồi quy tuyến tính trên toàn bộ giá trị đặc trưng và mục tiêu.
Vì mục tiêu là đánh giá mối quan hệ giữa đặc trưng và mục tiêu, nên không cần chia dữ liệu thành tập huấn luyện và kiểm tra.
X và y đã được nạp sẵn cho bạn như sau:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Bài tập này là một phần của khóa học
Học có giám sát với scikit-learn
Hướng dẫn bài tập
- Import
LinearRegression. - Khởi tạo một mô hình hồi quy tuyến tính.
- Dự đoán giá trị doanh số bằng
X, lưu vàopredictions.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])