def main(options, split=.9): output_file_1 = options.get_input_path("split_training.csv") output_file_2 = options.get_input_path("split_test.csv") shuffle_split(options.review, output_file_1, output_file_2) return output_file_1, output_file_2
def test(): with open('test.txt', 'w') as f: for i in range(100): f.write("{0}\n".format(i)) shuffle_split('test.txt', 'test1.txt', 'test2.txt', 0.9)