Spark之獲取GBT二分類函數的概率值

  在Spark中,GBT(Gradient Boost Trees,提升樹)函數用於實現機器學習中的提升樹算法,目前僅支持二分類算法。筆者在實際工作中需要獲得其預測的概率值,無奈該函數沒有相應的方法。   經過筆者幾天的奮鬥,終於找到了解決之道。下面將分享在Spark中如何獲取GBT二分類函數的概率值的思路。   首先,查看GBT函數的Scala源代碼,其中的predict函數如下:      
相關文章
相關標籤/搜索