擅长:python、mysql、java
<pre><code># We saved all the K Fold samples in different list then we access to this throught [i]
from sklearn.model_selection import KFold
import numpy as np
import pandas as pd
kf = KFold(n_splits=4)
X = np.array([[1, 2], [3, 4], [1, 2], [3, 4]])
Y = np.array([0,0,0,1])
Y=Y.reshape(4,1)
X=pd.DataFrame(X)
Y=pd.DataFrame(Y)
X_train_base=[]
X_test_base=[]
Y_train_base=[]
Y_test_base=[]
for train_index, test_index in kf.split(X):
X_train, X_test = X.iloc[train_index,:], X.iloc[test_index,:]
Y_train, Y_test = Y.iloc[train_index,:], Y.iloc[test_index,:]
X_train_base.append(X_train)
X_test_base.append(X_test)
Y_train_base.append(Y_train)
Y_test_base.append(Y_test)
print(X_train_base[0])
print(Y_train_base[0])
print(X_train_base[1])
print(Y_train_base[1])
</code></pre>