from sklearn.model_selection import KFold
import numpy as np
import os
withopen("./dataset/all.txt","r")as f:
lines =[]for line in f.readlines():
line = line.strip('\n')#去掉列表中每一个元素的换行符
lines.append(line)# print(line)
num =1
X = np.arange(1549)# y = np.random.choice([1,2],12,p=[0.4,0.6])
kf = KFold(n_splits=5,shuffle=True,random_state=0)# 初始化KFoldfor train_index , test_index in kf.split(X):# 调用split方法切分数据withopen("./dataset/train.txt","w")as ft:for i in train_index:# print(i)
ft.write(lines[i]+'\n')withopen("./dataset/val.txt","w")as fv:for i in test_index:# print(i)
fv.write(lines[i]+'\n')# os.system('nohup python train%d.py > out%d.log'%(num,num))
os.system('python train%d.py > out%d.log'%(num,num))
num = num +1
最主要的一句代码是:import osos.system('python train.py')下面是我自己写的进行五折验证的代码。from sklearn.model_selection import KFoldimport numpy as npimport oswith open("./dataset/all.txt", "r") as f: lines = [] for line in f.readlines(): line = line.strip(