2017-03-27 27 views
1

我訓練隨機森林分類樹參數的數量:Graphlab - 如何設置隨機森林分類

model = gl.random_forest_classifier.create(train, target = 'label',row_subsample = 0.5, column_subsample = 0.75, validation_set=validation, metric="auc", max_iterations=10, max_depth = 15) 

如何設置參數樹木的數量?這是一個二進制分類問題,文檔中提到:

max_iterations:要執行的最大迭代次數。對於具有K類的多類分類,每次迭代將創建K-1樹。

回答

1

「的num_trees關鍵字參數已棄用,請使用max_iterations說法代替。僅供num_trees任何值都將代替max_iterations使用。」

因此,調整max_iterations也意味着調整的樹木參數的數量。

+0

您能否爲此提供任何參考? – Tarun

+1

@Tarun \t 您可以在位置\ Lib \ site-packages \ graphlab \ toolkits \ classifier中打開random_forest_classifier.py的源代碼,行號爲653-655。 –