1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Machine Learning cho Marketing với Python
Connected

Bài tập

Trung bình phân khúc bằng NMF

Cuối cùng, bạn sẽ trực quan hóa các giá trị trung bình của lời giải 3 phân khúc được xây dựng bằng NMF và so sánh với K-means. Ở đây bạn sẽ trích xuất ma trận đặc trưng W, dùng để lấy phân khúc “cứng” bằng cách chọn tên cột (phân khúc) có giá trị lớn nhất trong ma trận này cho mỗi khách hàng.

Chúng tôi đã nạp thư viện pandas là pd và thư viện seaborn là sns. Bộ dữ liệu thô wholesale đã được nhập, và một đối tượng NMF 3 phân khúc đã được fit sẵn là nmf. Bộ dữ liệu components đã được nạp dưới dạng DataFrame của pandas.

Hướng dẫn

100 XP
  • Tạo ma trận W bằng cách truyền các giá trị đã được transform làm dữ liệu, và chỉ số của components làm tên cột.
  • Gán giá trị phân khúc bằng cách chọn tên cột có giá trị tương ứng lớn nhất.
  • Tính giá trị trung bình theo cột cho từng phân khúc.
  • Vẽ heatmap thể hiện các giá trị trung bình.