python脚本划分训练集、验证集、测试集
import os
import shutil
import random
# 保证随机可复现
random.seed(0)
def mk_dir(file_path):
if os.path.exists(file_path):
# 如果文件夹存在,则先删除原文件夹在重新创建
shutil.rmtree(file_path)
os.makedirs(file_path)
def split_data(file_path,new_file_path,t.
原创
2021-11-06 18:48:23 ·
885 阅读 ·
0 评论