pdb-gtl/make_test_set.py

   1 #!/usr/bin/python
   2
   3 import os, random, shutil
   4
   5 raw= 'rawpat_files_merged'
   6 out = raw + '_test'
   7 ranks = os.listdir(raw)
   8 for rank in ranks:
   9         sourcedir = raw + '/'+ rank
  10         outdir = out + '/' + rank
  11         try:
  12                 os.makedirs(outdir)
  13         except:
  14                 pass
  15
  16         plays = os.listdir(sourcedir)
  17         le = int(len(plays):qa  5)
  18         test_p = random.sample(plays, le)
  19
  20         for p in test_p:
  21                 shutil.move(sourcedir + '/' + p, outdir + '/' + p)
  22