2012-01-27 44 views
16

Bạn có biết thư viện tốt về việc tăng cường học máy bằng cây cỏ không?Thư viện cho cây tăng cường độ dốc

tốt:

  • với các thuật toán tốt như AdaBoost, TreeBoost, AnyBoost, LogitBoost, vv
  • với phân loại yếu cấu hình
  • khả năng của cả hai phân loại và dự đoán (hồi quy)
  • với tất cả các loại tín hiệu được phép: số, danh mục hoặc văn bản miễn phí
  • C/C++ hoặc Python
  • mở ra ource

Cho đến nay tôi đã tìm thấy http://www.multiboost.org/home có vẻ tốt. Nhưng tôi tự hỏi nếu có các thư viện khác?

Trả lời

0

Cá nhân tôi thích chạy weka (là java) bằng cách sử dụng mô-đun quy trình con trăn. Tuy nhiên, đồng nghiệp của tôi thường xuyên sử dụng:

  • orange - được đề xuất cho tôi là bộ công cụ học máy tốt nhất cho Python.
  • opencv - là C++, nhưng có các ràng buộc python. Thư viện này ban đầu được phát triển cho Computer Vision nhưng có nhiều thuật toán ML được triển khai (bao gồm boosting).
+0

Weka có vẻ tốt, cảm ơn. Màu da cam dường như không hỗ trợ cây dốc (afaik chỉ hỗ trợ một cây). OpenCV hỗ trợ thúc đẩy cây nhưng có vẻ khá chuyên về thị giác máy tính (chúng tôi cần nó nhiều hơn để phân tích văn bản nên tôi không chắc liệu nó có thể sử dụng được) hay không. Cảm ơn bạn! –

3

Những điều này không nhất thiết phải đáp ứng tất cả sở thích của bạn, nhưng cũng có:

  • Treenet một thương mại và mở rộng thực hiện ban đầu Jerome Friedman. Không phải nguồn mở nhưng chúng tôi đã tìm thấy nó hoạt động khá tốt
  • R gbm package cho các cây được tô đậm đặc biệt.
+0

R gbm có vẻ tốt. Chúng tôi sẽ thử nó. Cảm ơn. –

14

Nếu bạn đang tìm kiếm một phiên bản python, phiên bản mới nhất của scikit-learn có các cây hồi quy được tăng cường để phân loại và hồi quy (docs).

Nó tương tự như gói GBM R - GBM là nhanh hơn cho (bình phương nhỏ nhất) wheres hồi quy scikit-học của thực hiện là nhanh hơn ở bài kiểm tra thời gian và khi số điện thoại của các tính năng> 1000.

2

Tôi muốn giới thiệu xgboost (mà không tồn tại vào thời điểm câu hỏi được hỏi), là một gói mã nguồn mở R/python.

Hiện tại, nó là một trong những phương pháp tăng cường độ dốc nhanh nhất của cây, cho phép hồi quy/phân loại, hỗ trợ ma trận thưa thớt ...

Các vấn đề liên quan