拆分txt文件

最新推荐文章于 2024-09-05 13:55:59 发布

泡脚早睡早饭

最新推荐文章于 2024-09-05 13:55:59 发布

阅读量214

点赞数

文章标签： python

本文链接：https://blog.csdn.net/qq_43312665/article/details/129913355

版权

该代码段用于从anot_sort.txt文件中按75%的比例划分数据，创建训练集(train.txt)和测试集(test.txt)。首先计算文件的总行数，然后将前75%的行写入训练集，剩余的行写入测试集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

if __name__ == '__main__':
    fa = open('anot_sort.txt', 'r')
    count = len(fa.readlines())
    fa.close()
    train_n=int(count*0.75) #测试集正样本个数
    test_n=count-train_n #训练集正样本个数
    print(count,train_n,test_n)
    fa = open('anot_sort.txt', 'r')
    ftrain=open('train.txt', 'w')
    for line in fa.readlines()[:train_n]: # 训练集取前train_n
        ftrain.write(line)
    ftrain.close()
    fa.close()
    fa = open('anot_sort.txt', 'r')
    ftest = open('test.txt', 'w')
    for line in fa.readlines()[train_n:]:  # 测试集取剩余的
        ftest.write(line)
    ftest.close()
    fa.close()