1029 add octuple

2025-10-29 17:14:33 +08:00
parent b493ede479
commit e16c84aab2
22 changed files with 1135 additions and 62 deletions
--- a/Amadeus/symbolic_encoding/augmentor.py
+++ b/Amadeus/symbolic_encoding/augmentor.py
@ -22,6 +22,8 @@ class Augmentor:
      self.chord_idx = self.feature_list.index('chord')
  
  def _get_shift(self, segment):
+    if self.encoding_scheme == 'oct':
+      return 0
    # the pitch vocab has ignore token in 0 index
    if self.encoding_scheme == 'cp' or self.encoding_scheme == 'nb':
      pitch_mask = segment != 0