Chia dữ liệu thành huấn luyện và kiểm tra
Bước cuối cùng trước khi chuyển sang xây dựng mô hình hồi quy! Tại đây, bạn sẽ lần lượt xác định tên của biến mục tiêu và các cột đặc trưng, trích xuất dữ liệu, rồi chia thành tập huấn luyện và kiểm tra.
Thư viện pandas và numpy đã được nạp với bí danh lần lượt là pd và np. Các đặc trưng đầu vào đã được nhập dưới dạng bộ dữ liệu features, và biến mục tiêu bạn đã xây dựng ở bài trước đã được nhập sẵn cho bạn dưới tên Y.
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ột định danh khách hàng dưới dạng một danh sách.
- Chọn các tên cột đặc trưng, loại trừ cột định danh khách hàng.
- Trích xuất các đặc trưng thành
X. - Chia dữ liệu thành huấn luyện và kiểm tra bằng hàm
train_test_split().
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 customer identifier column name as a list
custid = ['___']
# Select feature column names excluding customer identifier
cols = [col for col in features.___ if col not in ___]
# Extract the features as `X`
X = features[___]
# Split data to training and testing
___, test_X, train_Y, ___ = ___(X, Y, test_size=0.25, random_state=99)