数据预处理(一):源数据的处理与写入

这篇博客主要介绍了如何整合A类和B类数据的训练集、验证集和测试集。首先,将源数据放置在"data"文件夹中,处理后的数据保存在"match_data"文件夹。使用特定的处理函数对每一句数据进行操作,并通过循环遍历数据文件,将不同类别的数据分别写入新的文件。最后,博主提到后续笔记会讲解如何将这些整合后的数据转换为BERT输入格式。
摘要由CSDN通过智能技术生成
  • 需求:将所有文件下的A类数据的训练集、验证集和测试集分别整合到一起,B类数据的训练集、验证集和测试集分别整合到一起。

  • 将要处理的数据文件放至data文件下,数据处理完成后存放至match_data文件下。
    在这里插入图片描述
    数据格式如下:
    在这里插入图片描述

  • 导入所需要的包

import os
import json
from tqdm import tqdm
  • 文件路径及目录相关操作
data_path = "data/" # 根路径
path_list = os.listdir(data_path) # 获取根路径下的文件目录
print(path_list)
  • 分别定义用于存放各类数据的列表
# 分别定义用于存放各类数据的列表
train_data_a = []
val_data_a = []
train_data_b = []
val_data_b = []
test_data_a = []
test_data_b = []
  • 对单句进行处理
# 对单句进行处理
def process(sentence):
    sentence = sentence.replace(" ", "") # 去空格操作
    sentence = sentence.lower() # 小写化操作
    return sentence
  • 将数据写入新文件的操作
# 将数据写入新文件操作
def write_data(path, data):
    with open(path, "w", encoding
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值