INNER CODE UNIT · Python
__predict
jisungk/deepjazz · generator.py:38
def __predict(model, x, indices_val, diversity):
preds = model.predict(x, verbose=0)[0]
next_index = __sample(preds, diversity)
next_val = indices_val[next_index]
return next_val
''' Helper function which uses the given model to generate a grammar sequence
from a given corpus, indices_val (mapping), abstract_grammars (list),
and diversity floating point value. '''
def __generate_grammar(model, corpus, abstract_grammars, values, val_indices,
indices_val, max_len, max_tries, diversity):
curr_grammar = ''
# np.random.randint is exclusive to high
start_index = np.random.randint(0, len(corpus) - max_len)
sentence = corpus[start_index: start_index + max_len] # seed
running_length = 0.0
while running_length <= 4.1: # arbitrary, from avg in input file