def testNumClasses(self):
self.assertEqual(
14, melody_encoder_decoder.MelodyOneHotEncoding(60, 72).num_classes)
self.assertEqual(
130, melody_encoder_decoder.MelodyOneHotEncoding(0, 128).num_classes)
self.assertEqual(
3, melody_encoder_decoder.MelodyOneHotEncoding(60, 61).num_classes)
def testInit(self):
melody_encoder_decoder.MelodyOneHotEncoding(0, 128)
with self.assertRaises(ValueError):
melody_encoder_decoder.MelodyOneHotEncoding(-1, 12)
with self.assertRaises(ValueError):
melody_encoder_decoder.MelodyOneHotEncoding(60, 129)
with self.assertRaises(ValueError):
melody_encoder_decoder.MelodyOneHotEncoding(72, 72)
def testDecodeEvent(self): enc = melody_encoder_decoder.MelodyOneHotEncoding(60, 72) self.assertEqual(63, enc.decode_event(5)) self.assertEqual(60, enc.decode_event(2)) self.assertEqual(71, enc.decode_event(13)) self.assertEqual(NO_EVENT, enc.decode_event(0)) self.assertEqual(NOTE_OFF, enc.decode_event(1))
def setUp(self):
super().setUp()
self.min_note = 60
self.max_note = 72
self.transpose_to_key = 0
self.med = encoder_decoder.OneHotEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(self.min_note,
self.max_note))
def testEncodeEvent(self):
enc = melody_encoder_decoder.MelodyOneHotEncoding(60, 72)
self.assertEqual(2, enc.encode_event(60))
self.assertEqual(13, enc.encode_event(71))
self.assertEqual(0, enc.encode_event(NO_EVENT))
self.assertEqual(1, enc.encode_event(NOTE_OFF))
with self.assertRaises(ValueError):
enc.encode_event(-3)
with self.assertRaises(ValueError):
enc.encode_event(59)
with self.assertRaises(ValueError):
enc.encode_event(72)
def __init__(self,
min_note=DEFAULT_MIN_NOTE,
max_note=DEFAULT_MAX_NOTE) -> None:
# self._note_counts = OrderedDict()
# self._note_pieces = {}
# self._note_index = {}
# self._index_note = {}
self._songs = []
self._song_count = 0
self._min_note = min_note
self._max_note = max_note
self._encoder_decoder = encoder_decoder.OneHotEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(min_note, max_note)
) # min_note=DEFAULT_MIN_NOTE, max_note=DEFAULT_MAX_NOTE
# Additional labels are NO_EVENT = 0 and NOTE_OFF = 1
assert (self._encoder_decoder.input_size, max_note - min_note + 2)
assert (self._encoder_decoder.num_classes, max_note - min_note + 2)
def __init__(self,
split_in_bar_chunks=4,
min_note=DEFAULT_MIN_NOTE,
max_note=DEFAULT_MAX_NOTE,
steps_per_quarter=DEFAULT_STEPS_PER_QUARTER) -> None:
self._song_parts_lead = []
self._song_parts_accomp = []
self._min_note = min_note
self._max_note = max_note
self._split_in_bar_chunks = split_in_bar_chunks
self._steps_per_quarter = steps_per_quarter
self.stats = dict()
self.midi_names = []
self.splits_per_midi = []
self.vocab = [-2, -1] + list(range(self._min_note, self._max_note))
self.counter = 0
self._encoder_decoder = encoder_decoder.OneHotEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(min_note, max_note))
# Additional labels are NO_EVENT = 0 and NOTE_OFF = 1
assert (self._encoder_decoder.input_size, max_note - min_note + 2)
assert (self._encoder_decoder.num_classes, max_note - min_note + 2)
def testDefaultEvent(self):
self.assertEqual(
NO_EVENT,
melody_encoder_decoder.MelodyOneHotEncoding(60, 72).default_event)
def testCustomRange(self):
med = encoder_decoder.LookbackEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(min_note=24, max_note=36))
self.assertEqual(med.input_size, 49)
self.assertEqual(med.num_classes, 16)
melody_events = ([24, NO_EVENT, 25, 35, NOTE_OFF] + [NO_EVENT] * 11 +
[24, NOTE_OFF] + [NO_EVENT] * 14 +
[24, NOTE_OFF, 25, 34])
melody = melodies_lib.Melody(melody_events)
melody_indices = [0, 1, 2, 3, 4, 16, 17, 32, 33, 34, 35]
expected_inputs = [
# 24, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# NO_EVENT, lookbacks = (NO_EVENT, NO_EVENT)
[1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# 25, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# 35, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0],
# NOTE_OFF, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0],
# 24, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, 1.0, 1.0, 0.0],
# NOTE_OFF, lookbacks = (25, NO_EVENT)
[0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, 1.0, 0.0, 0.0],
# 24, lookbacks = (NOTE_OFF, NO_EVENT)
[0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0],
# NOTE_OFF, lookbacks = (NO_EVENT, 25)
[0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 0.0],
# 25, lookbacks = (NO_EVENT, 35)
[0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 1.0],
# 34, lookbacks = (NO_EVENT, NOTE_OFF)
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0]
]
expected_labels = [2, 15, 3, 13, 1, 14, 1, 15, 14, 15, 12]
melodies = [melody, melody]
full_length_inputs_batch = med.get_inputs_batch(melodies, True)
for i, melody_index in enumerate(melody_indices):
partial_melody = melodies_lib.Melody(melody_events[:melody_index])
self.assertListEqual(full_length_inputs_batch[0][melody_index],
expected_inputs[i])
self.assertListEqual(full_length_inputs_batch[1][melody_index],
expected_inputs[i])
softmax = [[[0.0] * med.num_classes]]
softmax[0][0][expected_labels[i]] = 1.0
med.extend_event_sequences([partial_melody], softmax)
self.assertEqual(list(partial_melody)[-1], melody_events[melody_index])
self.assertListEqual(
[expected_inputs[-1:], expected_inputs[-1:]],
med.get_inputs_batch(melodies))
def testDefaultRange(self):
med = encoder_decoder.LookbackEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(48, 84))
self.assertEqual(med.input_size, 121)
self.assertEqual(med.num_classes, 40)
melody_events = ([48, NO_EVENT, 49, 83, NOTE_OFF] + [NO_EVENT] * 11 +
[48, NOTE_OFF] + [NO_EVENT] * 14 +
[48, NOTE_OFF, 49, 82])
melody = melodies_lib.Melody(melody_events)
melody_indices = [0, 1, 2, 3, 4, 16, 17, 32, 33, 34, 35]
expected_inputs = [
# 48, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# NO_EVENT, lookbacks = (NO_EVENT, NO_EVENT)
[1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# 49, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 0.0],
# 83, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0],
# NOTE_OFF, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0],
# 48, lookbacks = (NO_EVENT, NO_EVENT)
[0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, 1.0, 1.0, 0.0],
# NOTE_OFF, lookbacks = (49, NO_EVENT)
[0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, 1.0, 0.0, 0.0],
# 48, lookbacks = (NOTE_OFF, NO_EVENT)
[0.0, 0.0,
1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0],
# NOTE_OFF, lookbacks = (NO_EVENT, 49)
[0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 0.0],
# 49, lookbacks = (NO_EVENT, 83)
[0.0, 0.0,
0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
1.0, 1.0, -1.0, -1.0, -1.0, 0.0, 1.0],
# 82, lookbacks = (NO_EVENT, NOTE_OFF)
[0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0,
1.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 1.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
-1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0]
]
expected_labels = [2, 39, 3, 37, 1, 38, 1, 39, 38, 39, 36]
melodies = [melody, melody]
full_length_inputs_batch = med.get_inputs_batch(melodies, True)
for i, melody_index in enumerate(melody_indices):
print(i)
partial_melody = melodies_lib.Melody(melody_events[:melody_index])
self.assertListEqual(full_length_inputs_batch[0][melody_index],
expected_inputs[i])
self.assertListEqual(full_length_inputs_batch[1][melody_index],
expected_inputs[i])
softmax = [[[0.0] * med.num_classes]]
softmax[0][0][expected_labels[i]] = 1.0
med.extend_event_sequences([partial_melody], softmax)
self.assertEqual(list(partial_melody)[-1], melody_events[melody_index])
self.assertListEqual(
[expected_inputs[-1:], expected_inputs[-1:]],
med.get_inputs_batch(melodies))
out_file1 = mel_path + 'mel1_out.mid'
out_file2 = mel_path + 'mel2_out.mid'
out_file1_trans = mel_path + 'mel1_trans_out.mid'
out_file1_pred = mel_path + 'mel1_pred_out.mid'
# FOR IDEAS OF USING THE OUTPUT (hot encodings) DATA FROM THIS, SEE EVENTUALLY magenta.models.shared.eventss_rnn_model.py
melody1 = melodies_lib.midi_file_to_melody(input_file1)
seq = melody1.to_sequence()
midi_io.sequence_proto_to_midi_file(seq, out_file1)
min_note = 60
max_note = 72
transpose_to_key = 2
mel_encoder = encoder_decoder.OneHotEventSequenceEncoderDecoder(
melody_encoder_decoder.MelodyOneHotEncoding(min_note, max_note)) # min_note=DEFAULT_MIN_NOTE, max_note=DEFAULT_MAX_NOTE
# Additional labels are NO_EVENT = 0 and NOTE_OFF = 1
assert(mel_encoder.input_size, max_note - min_note + 2)
assert(mel_encoder.num_classes, max_note - min_note + 2)
# squeeze midi into octaves determined by min_note and max_note and transposes to key = 0 => C major / A minor
melody1.squash(
min_note,
max_note,
transpose_to_key)
inputs, labels = mel_encoder.encode(melody1)
print(inputs)
print(labels)