def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(name='cqt', shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(name='chord_idx', shape=(None, ), dtype='int32')
learning_rate = optimus.Input(name='learning_rate', shape=None)
margin = optimus.Input(name='margin', shape=None)
margin_weight = optimus.Input(name='margin_weight', shape=None)
nll_weight = optimus.Input(name='nll_weight', shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(name='layer0',
input_shape=input_data.shape,
weight_shape=(12, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(name='layer1',
input_shape=layer0.output.shape,
weight_shape=(16, None, 7, 15),
act_type='relu')
layer2 = optimus.Conv3D(name='layer2',
input_shape=layer1.output.shape,
weight_shape=(20, None, 6, 15),
act_type='relu')
layer3 = optimus.Affine(name='layer3',
input_shape=layer2.output.shape,
output_shape=(None, 512),
act_type='relu')
chord_classifier = optimus.Affine(name='chord_classifier',
input_shape=layer3.output.shape,
output_shape=(None, VOCAB),
act_type='sigmoid')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
nll = optimus.NegativeLogLikelihood(name="nll", weighted=True)
likelihood_margin = optimus.LikelihoodMargin(name="likelihood_margin",
mode='l1',
weighted=True)
# likelihood_margin = optimus.NLLMargin(
# name="likelihood_margin",
# mode='l2',
# weighted=True)
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input), (layer0.output, layer1.input),
(layer1.output, layer2.input), (layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, likelihood_margin.likelihood),
(chord_idx, likelihood_margin.target_idx),
(margin, likelihood_margin.margin),
(margin_weight, likelihood_margin.weight),
(chord_classifier.output, nll.likelihood), (chord_idx, nll.target_idx),
(nll_weight, nll.weight)
])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights), (learning_rate, layer0.bias),
(learning_rate, layer1.weights), (learning_rate, layer1.bias),
(learning_rate, layer2.weights), (learning_rate, layer2.bias),
(learning_rate, layer3.weights), (learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)
])
trainer = optimus.Graph(name=GRAPH_NAME,
inputs=[
input_data, chord_idx, margin, learning_rate,
margin_weight, nll_weight
],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[nll, likelihood_margin],
updates=update_manager.connections)
optimus.random_init(chord_classifier.weights)
optimus.random_init(chord_classifier.bias)
validator = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, margin, margin_weight, nll_weight],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[nll, likelihood_margin])
posterior = optimus.Output(name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input), (layer0.output, layer1.input),
(layer1.output, layer2.input), (layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)
])
predictor = optimus.Graph(name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
stash = biggie.Stash(args.training_file)
stream = S.minibatch(D.create_uniform_quality_stream(stash,
TIME_DIM,
vocab_dim=VOCAB),
batch_size=50)
driver = optimus.Driver(graph=trainer,
name=args.trial_name,
output_directory=args.model_directory)
hyperparams = {
learning_rate.name: LEARNING_RATE,
margin_weight.name: MARGIN_WEIGHT,
nll_weight.name: NLL_WEIGHT,
margin.name: MARGIN
}
driver.fit(stream, hyperparams=hyperparams, **DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(
name='cqt',
shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(
name='chord_idx',
shape=(None,),
dtype='int32')
learning_rate = optimus.Input(
name='learning_rate',
shape=None)
margin = optimus.Input(
name='margin',
shape=None)
margin_weight = optimus.Input(
name='margin_weight',
shape=None)
nll_weight = optimus.Input(
name='nll_weight',
shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(
name='layer0',
input_shape=input_data.shape,
weight_shape=(12, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(
name='layer1',
input_shape=layer0.output.shape,
weight_shape=(16, None, 7, 15),
act_type='relu')
layer2 = optimus.Conv3D(
name='layer2',
input_shape=layer1.output.shape,
weight_shape=(20, None, 6, 15),
act_type='relu')
layer3 = optimus.Affine(
name='layer3',
input_shape=layer2.output.shape,
output_shape=(None, 512),
act_type='relu')
chord_classifier = optimus.Softmax(
name='chord_classifier',
input_shape=layer3.output.shape,
n_out=VOCAB,
act_type='linear')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
nll = optimus.NegativeLogLikelihood(
name="nll",
weighted=True)
likelihood_margin = optimus.LikelihoodMargin(
name="likelihood_margin",
mode='l1',
weighted=True)
# likelihood_margin = optimus.NLLMargin(
# name="likelihood_margin",
# mode='l2',
# weighted=True)
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, likelihood_margin.likelihood),
(chord_idx, likelihood_margin.target_idx),
(margin, likelihood_margin.margin),
(margin_weight, likelihood_margin.weight),
(chord_classifier.output, nll.likelihood),
(chord_idx, nll.target_idx),
(nll_weight, nll.weight)])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights),
(learning_rate, layer0.bias),
(learning_rate, layer1.weights),
(learning_rate, layer1.bias),
(learning_rate, layer2.weights),
(learning_rate, layer2.bias),
(learning_rate, layer3.weights),
(learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)])
trainer = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, margin,
learning_rate, margin_weight, nll_weight],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[nll, likelihood_margin],
updates=update_manager.connections)
optimus.random_init(chord_classifier.weights)
optimus.random_init(chord_classifier.bias)
validator = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, margin, margin_weight, nll_weight],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[nll, likelihood_margin])
posterior = optimus.Output(
name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)])
predictor = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
stash = biggie.Stash(args.training_file)
stream = S.minibatch(
D.create_uniform_quality_stream(stash, TIME_DIM, vocab_dim=VOCAB),
batch_size=50)
driver = optimus.Driver(
graph=trainer,
name=args.trial_name,
output_directory=args.model_directory)
hyperparams = {learning_rate.name: LEARNING_RATE,
margin_weight.name: MARGIN_WEIGHT,
nll_weight.name: NLL_WEIGHT,
margin.name: MARGIN}
driver.fit(stream, hyperparams=hyperparams, **DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(
name='cqt',
shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(
name='chord_idx',
shape=(None,),
dtype='int32')
learning_rate = optimus.Input(
name='learning_rate',
shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(
name='layer0',
input_shape=input_data.shape,
weight_shape=(12, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(
name='layer1',
input_shape=layer0.output.shape,
weight_shape=(16, None, 7, 15),
act_type='relu')
layer2 = optimus.Conv3D(
name='layer2',
input_shape=layer1.output.shape,
weight_shape=(20, None, 6, 15),
act_type='relu')
layer3 = optimus.Affine(
name='layer3',
input_shape=layer2.output.shape,
output_shape=(None, 512,),
act_type='relu')
chord_classifier = optimus.Softmax(
name='chord_classifier',
input_shape=layer3.output.shape,
n_out=VOCAB,
act_type='linear')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
chord_nll = optimus.NegativeLogLikelihood(
name="chord_nll")
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, chord_nll.likelihood),
(chord_idx, chord_nll.target_idx)])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights),
(learning_rate, layer0.bias),
(learning_rate, layer1.weights),
(learning_rate, layer1.bias),
(learning_rate, layer2.weights),
(learning_rate, layer2.bias),
(learning_rate, layer3.weights),
(learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)])
print "Building trainer"
trainer = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, learning_rate],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll],
updates=update_manager.connections)
optimus.random_init(chord_classifier.weights)
print "Building validator"
validator = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll])
posterior = optimus.Output(
name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)])
print "Building predictor"
predictor = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
print "Opening Data"
stash = biggie.Stash(args.training_file)
stream = S.minibatch(
D.create_uniform_quality_stream(stash, TIME_DIM),
batch_size=50,
functions=[FX.pitch_shift(), FX.map_to_chord_index(VOCAB)])
driver = optimus.Driver(
graph=trainer,
name=args.trial_name,
output_directory=args.model_directory)
hyperparams = {learning_rate.name: LEARNING_RATE}
print "...aaand we're off!"
driver.fit(stream, hyperparams=hyperparams, **DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(name='cqt', shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(name='chord_idx', shape=(None, ), dtype='int32')
learning_rate = optimus.Input(name='learning_rate', shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(name='layer0',
input_shape=input_data.shape,
weight_shape=(30, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(name='layer1',
input_shape=layer0.output.shape,
weight_shape=(50, None, 7, 15),
act_type='relu')
layer2 = optimus.Affine(name='layer2',
input_shape=layer1.output.shape,
output_shape=(
None,
1024,
),
act_type='relu')
layer3 = optimus.Affine(name='layer3',
input_shape=layer2.output.shape,
output_shape=(
None,
1024,
),
act_type='relu')
chord_classifier = optimus.Softmax(name='chord_classifier',
input_shape=layer3.output.shape,
n_out=VOCAB,
act_type='linear')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
chord_nll = optimus.NegativeLogLikelihood(name="chord_nll")
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input), (layer0.output, layer1.input),
(layer1.output, layer2.input), (layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, chord_nll.likelihood),
(chord_idx, chord_nll.target_idx)
])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights), (learning_rate, layer0.bias),
(learning_rate, layer1.weights), (learning_rate, layer1.bias),
(learning_rate, layer2.weights), (learning_rate, layer2.bias),
(learning_rate, layer3.weights), (learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)
])
trainer = optimus.Graph(name=GRAPH_NAME,
inputs=[input_data, chord_idx, learning_rate],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll],
updates=update_manager.connections)
for n in all_nodes:
optimus.random_init(n.weights)
optimus.random_init(n.bias)
validator = optimus.Graph(name=GRAPH_NAME,
inputs=[input_data, chord_idx],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll])
posterior = optimus.Output(name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input), (layer0.output, layer1.input),
(layer1.output, layer2.input), (layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)
])
predictor = optimus.Graph(name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
stash = biggie.Stash(args.training_file)
hyperparams = {learning_rate.name: LEARNING_RATE}
valid_idx = [0]
for q in range(1, 13):
valid_idx.append(q)
stream = S.minibatch(D.create_uniform_quality_stream(
stash, TIME_DIM, vocab_dim=VOCAB, valid_idx=valid_idx),
batch_size=BATCH_SIZE)
driver = optimus.Driver(graph=trainer,
name=args.trial_name + "_c%02d" % q,
output_directory=args.model_directory)
driver.fit(stream,
hyperparams=hyperparams,
max_iter=20000,
**DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(
name='cqt',
shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(
name='chord_idx',
shape=(None,),
dtype='int32')
learning_rate = optimus.Input(
name='learning_rate',
shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(
name='layer0',
input_shape=input_data.shape,
weight_shape=(30, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(
name='layer1',
input_shape=layer0.output.shape,
weight_shape=(50, None, 7, 15),
act_type='relu')
layer2 = optimus.Affine(
name='layer2',
input_shape=layer1.output.shape,
output_shape=(None, 1024,),
act_type='relu')
layer3 = optimus.Affine(
name='layer3',
input_shape=layer2.output.shape,
output_shape=(None, 1024,),
act_type='relu')
chord_classifier = optimus.Softmax(
name='chord_classifier',
input_shape=layer3.output.shape,
n_out=VOCAB,
act_type='linear')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
chord_nll = optimus.NegativeLogLikelihood(
name="chord_nll")
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, chord_nll.likelihood),
(chord_idx, chord_nll.target_idx)])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights),
(learning_rate, layer0.bias),
(learning_rate, layer1.weights),
(learning_rate, layer1.bias),
(learning_rate, layer2.weights),
(learning_rate, layer2.bias),
(learning_rate, layer3.weights),
(learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)])
trainer = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, learning_rate],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll],
updates=update_manager.connections)
for n in all_nodes:
optimus.random_init(n.weights)
optimus.random_init(n.bias)
validator = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll])
posterior = optimus.Output(
name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)])
predictor = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
stash = biggie.Stash(args.training_file)
hyperparams = {learning_rate.name: LEARNING_RATE}
valid_idx = [0]
for q in range(1, 13):
valid_idx.append(q)
stream = S.minibatch(
D.create_uniform_quality_stream(stash, TIME_DIM, vocab_dim=VOCAB,
valid_idx=valid_idx),
batch_size=BATCH_SIZE)
driver = optimus.Driver(
graph=trainer,
name=args.trial_name + "_c%02d" % q,
output_directory=args.model_directory)
driver.fit(
stream, hyperparams=hyperparams, max_iter=20000, **DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
def main(args):
# 1.1 Create Inputs
input_data = optimus.Input(
name='cqt',
shape=(None, 1, TIME_DIM, 252))
chord_idx = optimus.Input(
name='chord_idx',
shape=(None,),
dtype='int32')
learning_rate = optimus.Input(
name='learning_rate',
shape=None)
limiter_weight = optimus.Input(
name='limiter_weight',
shape=None)
likelihood_threshold = optimus.Input(
name='likelihood_threshold',
shape=None)
# 1.2 Create Nodes
layer0 = optimus.Conv3D(
name='layer0',
input_shape=input_data.shape,
weight_shape=(12, 1, 9, 19),
pool_shape=(1, 3),
act_type='relu')
layer1 = optimus.Conv3D(
name='layer1',
input_shape=layer0.output.shape,
weight_shape=(16, None, 7, 15),
act_type='relu')
layer2 = optimus.Conv3D(
name='layer2',
input_shape=layer1.output.shape,
weight_shape=(20, None, 6, 15),
act_type='relu')
layer3 = optimus.Affine(
name='layer3',
input_shape=layer2.output.shape,
output_shape=(None, 512,),
act_type='relu')
chord_classifier = optimus.Softmax(
name='chord_classifier',
input_shape=layer3.output.shape,
n_out=VOCAB,
act_type='linear')
all_nodes = [layer0, layer1, layer2, layer3, chord_classifier]
# 1.1 Create Losses
chord_nll = optimus.NegativeLogLikelihood(
name="chord_nll")
max_likelihood = optimus.Max(
name="max_likelihood")
# 2. Define Edges
trainer_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, chord_nll.likelihood),
(chord_idx, chord_nll.target_idx),
(chord_classifier.output, max_likelihood.input),
(limiter_weight, max_likelihood.weight),
(likelihood_threshold, max_likelihood.threshold)])
update_manager = optimus.ConnectionManager([
(learning_rate, layer0.weights),
(learning_rate, layer0.bias),
(learning_rate, layer1.weights),
(learning_rate, layer1.bias),
(learning_rate, layer2.weights),
(learning_rate, layer2.bias),
(learning_rate, layer3.weights),
(learning_rate, layer3.bias),
(learning_rate, chord_classifier.weights),
(learning_rate, chord_classifier.bias)])
print "Trainer"
trainer = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, learning_rate,
limiter_weight, likelihood_threshold],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll, max_likelihood],
updates=update_manager.connections)
optimus.random_init(chord_classifier.weights)
print "Validator"
validator = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data, chord_idx, limiter_weight, likelihood_threshold],
nodes=all_nodes,
connections=trainer_edges.connections,
outputs=[optimus.Graph.TOTAL_LOSS],
losses=[chord_nll, max_likelihood])
posterior = optimus.Output(
name='posterior')
predictor_edges = optimus.ConnectionManager([
(input_data, layer0.input),
(layer0.output, layer1.input),
(layer1.output, layer2.input),
(layer2.output, layer3.input),
(layer3.output, chord_classifier.input),
(chord_classifier.output, posterior)])
print "Predictor"
predictor = optimus.Graph(
name=GRAPH_NAME,
inputs=[input_data],
nodes=all_nodes,
connections=predictor_edges.connections,
outputs=[posterior])
# 3. Create Data
stash = biggie.Stash(args.training_file)
stream = S.minibatch(
D.create_uniform_quality_stream(stash, TIME_DIM, vocab_dim=VOCAB),
batch_size=50)
driver = optimus.Driver(
graph=trainer,
name=args.trial_name,
output_directory=args.model_directory)
hyperparams = {learning_rate.name: LEARNING_RATE,
likelihood_threshold.name: LIKELIHOOD_THRESHOLD,
limiter_weight.name: LIMITER_WEIGHT}
driver.fit(stream, hyperparams=hyperparams, **DRIVER_ARGS)
validator_file = path.join(driver.output_directory, args.validator_file)
optimus.save(validator, def_file=validator_file)
predictor_file = path.join(driver.output_directory, args.predictor_file)
optimus.save(predictor, def_file=predictor_file)
