擅长:python、mysql、java
<p>将<code>train_test_split</code>中的<code>stratify</code>参数设置为目标列。你知道吗</p>
<p><code>stratify</code>将确保每个类被平均分割。<a href="https://scikit-learn.org/stable/modules/cross_validation.html#stratified-k-fold" rel="nofollow noreferrer">Doc</a></p>
<p><code>X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, stratify=y)</code></p>