作者Lynnhan ()
看板Statistics
標題[程式] CART decision tree 的 python寫法
時間Wed Dec 16 12:46:32 2020
[軟體程式類別]: Python
[程式問題]: CART 決策樹 寫法
[軟體熟悉度]: 新手
[問題敘述]:
大大們好 最近在自學decision tree
學到CART的部分 需要寫一個python function去表示
題目先給定了TreeNode的寫法 要我自訂一個cart function
應該是要不斷的build recursively 但我完全卡住了...
[程式範例]:
Tree Node:
http://codepad.org/agsBXJro
Tree Node 解釋:
left: left subtree
right: right subtree
feature: index of feature to cut
cut: cutoff value c (<=c : left, and >c : right)
prediction: prediction at this node
cart:
http://pastie.org/p/65l3bnfnAYgSsqXBS9LaBJ
其中的sqsplit是我定義的function
http://codepad.org/vqlhnwh7
輸出值是f 代表x裡的某個dimension c代表cutoff value loss表示資料的square差
如果有需要補充的我隨時補
有大大可以幫我解答的話願意以400P答謝!
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 64.18.152.57 (美國)
※ 文章網址: https://webptt.com/m.aspx?n=bbs/Statistics/M.1608094000.A.A54.html