在划分数据集时,随机取得训练集,剩下的是测试集。这里有一个麻烦事是训练集是随机的,所以要怎么减呢。我的想法是用集合来减。
import random
lst1 = ['1','2','3','4','5','6'] # 假设这是总的
split_percent = 0.8
train = random.sample(lst1,int(split_percent*len(lst1)))
test = list(set(lst1)-set(train))
print('train:',train)
print('test:',test)
结果: