view notebooks/sensitivity_experiment_server.py @ 40:e6e10013e11c branch-tests

small edits

import numpy as np
import sys
sys.path.append('../')
import scripts.load_dataset as load_dataset

df = load_dataset.sample_dataset(csv_file=load_dataset.METADATA_FILE)
OUTPUT_FILES = load_dataset.OUTPUT_FILES
n_iters = 1
n = int(sys.argv[1])

#for n in range(n_iters):
if 1:
    print "iteration %d" % n
    load_dataset.OUTPUT_FILES = [output_file.split('.pickle')[0]+'_'+str(n)+'.pickle' for 
                                 output_file in OUTPUT_FILES]
    load_dataset.features_for_train_test_sets(df, write_output=True)