最主要的一句代码是:
import os
os.system('python train.py')
下面是我自己写的进行五折验证的代码。
from sklearn.model_selection import KFold
import numpy as np
import os
with open("./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) # 初始化KFold
for train_index , test_index in kf.split(X): # 调用split方法切分数据
with open("./dataset/train.txt","w") as ft:
for i in train_index:
# print(i)
ft.write(lines[i] + 'n')
with open("./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