def evaluate(items, outs, config, trial=None):
data, bpm_label, beat_label = items
beat_out, bpm_out = outs
eval_beat = EvaluateBeat("downbeat")
eval_bpm = EvaluateBpm()
beat_metrics = eval_beat.batch(beat_out, beat_label)
bpm_metrics = eval_bpm.local(bpm_out, bpm_label)
ret = {"beat": beat_metrics, "bpm": bpm_metrics}
return ret
def evaluate(items, outs, config, optimize=False):
beat, bpm = items
eval_bpm = EvaluateBpm()
bpm_metrics = eval_bpm.local(outs, bpm)
ret = {"bpm": bpm_metrics}
return ret