def genTrainingData(forTraining): x=[] y=[] for i in xrange(len(data_gen.bins)): seq_len = data_gen.bins[i] seq_count=countList[i] data, labels = data_gen.get_batch(seq_len,seq_count,forTraining, task) x+=[data] y+=[labels] return x,y
def genTestData(length, count): data, labels = data_gen.get_batch(length, count, False, task) return [data], [labels]