m@32: import numpy as np m@32: import sys m@32: sys.path.append('../') m@32: import scripts.load_dataset as load_dataset m@32: m@32: df = load_dataset.sample_dataset(csv_file=load_dataset.METADATA_FILE) m@32: OUTPUT_FILES = load_dataset.OUTPUT_FILES m@32: n_iters = 1 m@32: n = int(sys.argv[1]) m@32: m@32: #for n in range(n_iters): m@32: if 1: m@32: print "iteration %d" % n m@32: load_dataset.OUTPUT_FILES = [output_file.split('.pickle')[0]+'_'+str(n)+'.pickle' for m@32: output_file in OUTPUT_FILES] m@32: load_dataset.features_for_train_test_sets(df, write_output=True)