2015-11-17 67 views
1

我是新來的XGBOOST包,我想用它來處理多類分類問題。如何在R中使用xgboost包來解釋多類分類的輸出?

我現在有30個類(或標籤),我已經將它們轉換爲整數(0 - 29)。然後,我從數據集上的XGBOOST包運行xgboost()方法。建立模型後,我使用predict()方法來預測新的測試數據。最後,我得到了一個數字矢量作爲最終輸出。順便說一句,所有的數字都在0和1之間。

那麼現在我怎麼能解釋輸出呢?這是可能性嗎?但是,我怎麼知道哪一類應該關聯?

謝謝

回答

2

我剛剛得到它!輸出是100,000 * 30值的矩陣。因此,每30個數字對應一個特定實例,這意味着每個數字是該實例成爲標籤的概率。

例如:

矩陣[1]是習題(instance0 == label0)

矩陣[32]是習題(INSTANCE1 == label1的)

矩陣[33] =習題(instance1 == label3)