Các lô phụ thuộc một phần và tăng cường Gradient (gói GBM)


9

Có thể vẽ sơ đồ phụ thuộc một phần để hiển thị xác suất lớp và ước tính tác động của bộ dự đoán cho mô hình GBM không? Một cái gì đó tương tự partialPlottừ randomForestgói.

nhập mô tả hình ảnh ở đây

Theo bài viết này , một phần cốt truyện có thể thực hiện được với gbm.

Cảm ơn trước sự giúp đỡ của bạn.

Câu trả lời:


12

Vâng, nó thực sự dễ dàng. Kiểm tra ?plot.gbm. Việc sử dụng diễn ra như sau:

 plot(gbm.model, i.var = 1, lwd = 2, col = "blue", main = "")

và trông giống như nhập mô tả hình ảnh ở đây

i.varkiểm soát chỉ số của biến để vẽ. Bạn cũng có thể thực hiện hai (hoặc nhiều) cách sử dụng

 plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")

1
cảm ơn bạn, tôi thực sự đánh giá cao những nỗ lực và thời gian của bạn ở đây và trong blog của bạn.
Panos Kal.
Khi sử dụng trang web của chúng tôi, bạn xác nhận rằng bạn đã đọc và hiểu Chính sách cookieChính sách bảo mật của chúng tôi.
Licensed under cc by-sa 3.0 with attribution required.