Python Scikit學習 - 不想隨機分割使用train_test_split

Question

我使用Scikit學習train_test_split和隨機森林進行建模。 我已經在train_test_split的random_state中設置了固定的數字，比如說random_state = 1，但它每次運行都會產生不同的結果。我想要一個固定的結果來比較模型。 非常感謝。

Answer 1

你是什麼意思的「不同的結果」？我建議在將它傳遞給您的培訓代碼之前，打印一份簡短的培訓/測試集預覽。例如：

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25 random_state = 1) 
print(X_train[:10]) 

我可以確認這段代碼將在多次運行中產生相同的確切X_train集合。

但是，如果「不同的結果」表示訓練錯誤不同，那麼您可能需要確定您正在修復隨機森林分類器的隨機種子。