EL之RF(RFC):利用RF對多分類問題進行建模並評估(六分類+分層抽樣)
阿新 • • 發佈:2019-01-14
EL之RF(RFC):利用RF對多分類問題進行建模並評估(六分類+分層抽樣)
輸出結果
設計思路
核心程式碼
missCLassError = [] nTreeList = range(50, 2000, 50) for iTrees in nTreeList: depth = None maxFeat = 4 #try tweaking glassRFModel = ensemble.RandomForestClassifier(n_estimators=iTrees, max_depth=depth, max_features=maxFeat, oob_score=False, random_state=531) glassRFModel.fit(xTrain,yTrain) prediction = glassRFModel.predict(xTest) correct = accuracy_score(yTest, prediction) missCLassError.append(1.0 - correct)