PYTHON 生成并附带相应的训练集和测试集及其标签
发布时间
阅读量:
阅读量
描述
本研究旨在应对神经网络训练过程中,需对训练集与测试集进行随机划分的问题,以确保训练集能够更全面地涵盖测试集的均值与方差范围。
其中,
(1)数据存储于csv文件中,这是由于网络的输入形式为二维点云数据
(2)标签信息同样保存在csv文件中,该文件包含四列内容,第一列为对应数据文件的名称,第二、第三及第四列则为待预测的具体数值。
导入包
在本次操作过程中,需借助os库实现文件夹的创建,并用于确定标签文件的存储路径;同时利用csv库对csv格式的数据文件进行处理。此外,还需引用csv文件批量处理可以参考这一篇博客模块以完成相关功能,通过random库生成随机数值,并借助shutil库将csv文件复制至指定的训练集与测试集文件夹中。
import csv
import matplotlib.pyplot as plt
import numpy as np
import random
import os
import shutil
Note:本文没有注重效率问题,如果要兼顾效率的可能不适合
分割训练集和测试集
1. 创建训
全部评论 (0)
还没有任何评论哟~
