Dự đoán xác suất của các bài đánh giá phim
Trong bài này, bạn sẽ xây dựng logistic regression với bộ dữ liệu movies. Nhãn được lưu trong mảng y và các đặc trưng nằm trong X.
Huấn luyện mô hình trên dữ liệu huấn luyện. Thay vì dự đoán lớp, hãy dự đoán xác suất mỗi mẫu trong tập kiểm tra thuộc về từng trong hai lớp.
Các hàm logistic regression và chia train/test đã được import sẵn cho bạn.
Bài tập này là một phần của khóa học
Phân tích cảm xúc với Python
Hướng dẫn bài tập
- Chia dữ liệu thành tập huấn luyện và tập kiểm tra.
- Huấn luyện một mô hình logistic regression.
- Dự đoán xác suất cho lớp 0 và lớp 1 của dữ liệu kiểm tra. Lớp 0 nằm ở cột đầu tiên trong ma trận xác suất dự đoán, và lớp 1 nằm ở cột thứ hai.
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.
# Split into training and testing
X_train, X_test, y_train, y_test = ____(___, ___, test_size=0.2, random_state=321)
# Train a logistic regression
log_reg = ____.____
# Predict the probability of the 0 class
prob_0 = log_reg.____[:, ____]
# Predict the probability of the 1 class
prob_1 = log_reg.____[:, ____]
print("First 10 predicted probabilities of class 0: ", prob_0[:10])
print("First 10 predicted probabilities of class 1: ", prob_1[:10])