Tách đặc trưng và biến mục tiêu
Giờ bạn đã chia dữ liệu thành tập huấn luyện và kiểm tra, đây là lúc thực hiện bước cuối trước khi huấn luyện mô hình: tách các biến đặc trưng và biến mục tiêu thành các tập dữ liệu riêng. Bạn sẽ dùng danh sách tên cột đã được nạp sẵn.
Tập dữ liệu chính được nạp là telcom, và đã được chia thành tập huấn luyện và kiểm tra, lần lượt được nạp dưới dạng DataFrame của pandas là train và test. Các danh sách target và custid lần lượt chứa tên của biến mục tiêu và mã khách hàng. Bạn sẽ cần tạo danh sách cols với tên của các cột còn lại. Bạn có thể thoải mái khám phá các tập dữ liệu trong bảng điều khiển.
Bài tập này là một phần của khóa học
Machine Learning cho Marketing với Python
Hướng dẫn bài tập
- Lưu tên các cột của
telcomvào một danh sách, loại trừ tên biến mục tiêu và mã khách hàng. - Trích xuất đặc trưng và mục tiêu cho tập huấn luyện.
- Trích xuất đặc trưng và mục tiêu cho tập kiểm tra.
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.
# Store column names from `telcom` excluding target variable and customer ID
cols = [col for col in ___.columns if col not in ___ + target]
# Extract training features
train_X = train[___]
# Extract training target
train_Y = train[___]
# Extract testing features
test_X = test[___]
# Extract testing target
test_Y = test[___]