我一直在使用 sklearn 的随机森林,并尝试比较几个模型。然后我注意到即使使用相同的种子,随机森林也会给出不同的结果。我尝试了两种方式:random.seed(1234) 以及使用随机森林内置 random_state = 1234 在这两种情况下,我得到不可重复的结果。我错过了什么……?
# 1
random.seed(1234)
RandomForestClassifier(max_depth=5, max_features=5, criterion='gini', min_samples_leaf = 10)
# or 2
RandomForestClassifier(max_depth=5, max_features=5, criterion='gini', min_samples_leaf = 10, random_state=1234)
有任何想法吗?谢谢!!
编辑:添加我的代码的更完整版本
clf = RandomForestClassifier(max_depth=60, max_features=60, \
criterion='entropy', \
min_samples_leaf = 3, random_state=seed)
# As describe, I tried random_state in several ways, still diff results
clf = clf.fit(X_train, y_train)
predicted = clf.predict(X_test)
predicted_prob = clf.predict_proba(X_test)[:, 1]
fpr, tpr, thresholds = metrics.roc_curve(np.array(y_test), predicted_prob)
auc = metrics.auc(fpr,tpr)
print (auc)
编辑:已经有一段时间了,但我认为使用RandomState可能会解决问题。我自己还没有测试它,但如果你正在阅读它,它值得一试。此外,通常最好使用 RandomState 而不是 random.seed()。