機器學習實戰-決策樹

決策樹的第一個算法爲,計算給定數據集的香農熵 香農熵的計算公式都給出來了,下面貼一下代碼 還是比較好懂的,就是先創建一個字典,在字典裏面對每個類別新建一個鍵值對,值爲每個類別的樣本個數 之後分別計算概率啊什麼的 輸出如下 跟書上的答案是一樣的 我個人有一點點的疑問是對字典類型的自建函數 下面看看python文檔 dict.keys()              返回一個包含字典中鍵的列表 >>>
相關文章
相關標籤/搜索