Uniform Manifold Approximation and Projection (UMAP)

Giảm Chiều Dữ Liệu với R

Matt Pickard

Owner, Pickard Predictives, LLC

PCA, t-SNE và UMAP

So sánh PCA, t-SNE và UMAP

PCA, t-SNE và UMAP

So sánh PCA, t-SNE và UMAP

PCA, t-SNE và UMAP

So sánh PCA, t-SNE và UMAP

PCA, t-SNE và UMAP

So sánh PCA, t-SNE và UMAP

PCA, t-SNE và UMAP

So sánh PCA, t-SNE và UMAP

UMAP có các siêu tham số tương tự có thể tinh chỉnh.

Biểu đồ UMAP

library(embed)


set.seed(1234)
umap_df <- recipe(Attrition ~ ., data = attrition_df) %>% 
  step_normalize(all_predictors()) %>% 
  step_umap(all_predictors(), num_comp = 2) %>% 
  prep() %>% 
  juice()


umap_df %>%  
  ggplot(aes(x = UMAP1, y = UMAP2, color = Attrition)) +
  geom_point(alpha = 0.7)

UMAP: nghỉ việc của nhân viên

Biểu đồ UMAP về nghỉ việc của nhân viên

UMAP trong tidymodels

Tạo recipe

umap_recipe <-  recipe(Attrition ~ ., data = train) %>% 
  step_normalize(all_predictors()) %>% 
  step_umap(all_predictors(), num_comp = 4)

Tạo đặc tả mô hình

umap_lr_model <- linear_reg()

UMAP trong tidymodels

Tạo workflow

umap_lr_workflow <-  workflow() %>% 
  add_recipe(umap_recipe) %>% 
  add_model(umap_lr_model)

Fit workflow

umap_lr_fit <- umap_lr_workflow %>% 
  fit(data = train)

UMAP trong tidymodels

Đánh giá mô hình

predict_umap_df <- test %>% 
  bind_cols(predict = predict(umap_lr_fit, test))

rmse(predict_umap_df, Attrition, .pred_class)

Ayo berlatih!

Giảm Chiều Dữ Liệu với R