Bắt đầu ngayBắt đầu miễn phí

Dự đoán doanh thu phim

Hãy bắt đầu thử thách dự đoán doanh thu phim bằng cách xây dựng một mô hình hồi quy tuyến tính đơn giản để ước lượng log-doanh thu dựa trên đặc trưng 'budget'. Thước đo bạn sẽ dùng ở đây là RMSE (root mean squared error). Để tính bằng scikit-learn, bạn có thể dùng hàm mean_squared_error() từ module sklearn.metrics, rồi lấy căn bậc hai của nó bằng numpy.

Bộ dữ liệu movies đã được tải sẵn và chia thành tập huấn luyện và kiểm tra. Ngoài ra, các giá trị khuyết đã được thay bằng 0. Chúng tôi cũng đã tiêu chuẩn hóa đặc trưng đầu vào bằng StandardScaler(). Xem các khóa học của DataCamp về làm sạch dữ liệu và kỹ thuật đặc trưng nếu bạn muốn tìm hiểu thêm về tiền xử lý cho Machine Learning.

Bài tập này là một phần của khóa học

Ensemble Methods in Python

Xem khóa học

Hướng dẫn bài tập

  • Khởi tạo mô hình LinearRegression mặc định.
  • Tính các dự đoán trên tập kiểm tra.
  • Tính RMSE. Hàm mean_squared_error() cần hai đối số: y_test, tiếp theo là các dự đoán.

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.

# Build and fit linear regression model
reg_lm = ____
reg_lm.fit(X_train, y_train)

# Calculate the predictions on the test set
pred = ____

# Evaluate the performance using the RMSE
rmse = np.sqrt(____)
print('RMSE: {:.3f}'.format(rmse))
Chỉnh sửa và Chạy Mã