jazzsamba

Python API for the JazzSAMBA multitrack jazz-standards dataset.

Install the package, point it at a local JazzSAMBA/ directory (after unzipping the Zenodo archives into one folder), and load songs, takes, and stems.

Related repos

Install

pip install -e .
# later: pip install jazzsamba
# needs setuptools>=68 (or the included setup.py for older pip)

From the processing monorepo (submodule):

pip install -e packages/jazzsamba

Quick start

from jazzsamba import JazzSamba

ds = JazzSamba("/path/to/JazzSAMBA")
# or JazzSamba() if JAZZSAMBA_DIR / JAZZ_SAMBA_ROOT is set (or a nearby .env)
song = ds.song(0)
take = song.recording.take("preferred")
mixture = take.mixture_path()
bass = take.stem("bass")
print(bass.audio_path, bass.midi_path, bass.midi_is_gt, bass.is_derived)
for stem in take.stems(is_derived=False):
    ...

Iterate and filter (rows whose protocol song directory is missing are skipped):

for song in ds.songs(synchronous=False, genre="swing"):
    take = song.recording.take("preferred")
    extras = song.recording.extras()  # async only; not a Take

reference.url / youtube_id come from songs.csv (pointers only; commercial/YouTube audio is not in JazzSAMBA).

Object model

See docs/ER.md. Canonical layout and song- vs take-level fields are also in the JazzSAMBA dataset README.md (shipped on Zenodo).

JazzSamba(root)
  └── Song
        └── AsyncRecording | SyncRecording
              └── Take (preferred | alternate)
              └── extras/   # async only, not a Take

Prefer:

take = song.recording.take("preferred")
take.measure_sequence   # derived from annotations/bars.csv
take.solo_order         # derived from annotations/soloists.csv (instrument column)
take.annotations()

Dataset root layout

JazzSAMBA/
  songs.csv
  musicians.csv
  async/
    splits/{train,val,test}.txt
    songs/<NN-title>/{preferred,alternate,extras}/
  sync/
    splits/{train,val,test}.txt
    songs/<NN-title>/{preferred,alternate}/

ids = ds.split_ids("async_test")  # reads async/splits/test.txt

License

MIT — see LICENSE.