1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Unsupervised Learning bằng Python

Connected

Bài tập

Bản đồ t-SNE của thị trường chứng khoán

t-SNE cho ra hình trực quan rất tốt khi từng mẫu có thể được gắn nhãn. Trong bài này, bạn sẽ áp dụng t-SNE lên dữ liệu giá cổ phiếu của các công ty. Một biểu đồ scatter từ các đặc trưng t-SNE, được gắn nhãn theo tên công ty, sẽ cho bạn một “bản đồ” của thị trường chứng khoán! Các biến động giá cổ phiếu cho mỗi công ty có trong mảng normalized_movements (đã được chuẩn hóa sẵn). Danh sách companies cung cấp tên của từng công ty. PyPlot (plt) đã được import sẵn cho bạn.

Hướng dẫn

100 XP
  • Import TSNE từ sklearn.manifold.
  • Tạo một đối tượng TSNE tên model với learning_rate=50.
  • Áp dụng phương thức .fit_transform() của model lên normalized_movements. Gán kết quả cho tsne_features.
  • Chọn cột 0 và cột 1 của tsne_features.
  • Vẽ biểu đồ scatter cho các đặc trưng t-SNE xs và ys. Chỉ định thêm tham số alpha=0.5.
  • Mã dùng plt.annotate() để gắn nhãn từng điểm bằng tên công ty đã được viết sẵn, bạn chỉ cần nhấn Gửi để xem hình trực quan!