24c14ca346506f31d789db35bb1b0173b7d97ce6,magenta/music/encoder_decoder_test.py,OneHotEventSequenceEncoderDecoderTest,testEncode,#OneHotEventSequenceEncoderDecoderTest#,61

Before Change

                       [1.0, 0.0, 0.0],
                       [0.0, 0.0, 1.0]]
    expected_labels = [1, 0, 2, 0]
    expected_sequence_example = encoder_decoder.make_sequence_example(
        expected_inputs, expected_labels)
    self.assertEqual(sequence_example, expected_sequence_example)

  def testGetInputsBatch(self):
    event_sequences = [[0, 1, 0, 2, 0], [0, 1, 2]]

After Change

  def testEncode(self):
    events = [0, 1, 0, 2, 0]
    inputs, labels = self.enc.encode(events)
    expected_inputs = [[1.0, 0.0, 0.0],
                       [0.0, 1.0, 0.0],
                       [1.0, 0.0, 0.0],

Instances