smg-metrics 5.1.0

Objective evaluation metrics for Symbolic Music Generation

smg-metrics

Symbolic Music Generation Metrics — 53 objective evaluation metrics, zero config.

8 categories, 53 metrics, 21 papers/projects (1990–2026), fully typed & tested.

Category	Count	Latest source	Year
A. Single-file Quality	14	FGG / MusPy / XMusic	2025
B. Note-level Pairwise	5	Ou et al.	2025
C. Bar-level Pairwise	2	MuseMorphose	2023
D. Chord-level Pairwise	2	FGG / Wang et al. ISMIR	2025/2020
E. Distribution-level	5	SongMASS	2020
F. Advanced	14	Text2midi	2025
G. Structural	4	MuseTok	2026
H. Rhythmic/Temporal	7	D3PIA	2026

Quick Start

pip install smg-metrics

from smg_metrics import single_file, single_file_rhythmic, pair_eval, compute_ook

# Single-file quality (13 MusPy metrics)
quality = single_file("generated.mid")
print(quality.pce, quality.ebr, quality.gs)

# Out-of-Key percentage (FGG 2025)
ook = compute_ook("generated.mid")
print(f"OOK: {ook:.2f}%")

# Rhythmic metrics (4 D3PIA-style)
rhythm = single_file_rhythmic("generated.mid")
print(rhythm.mean_ioi, rhythm.rhythmic_density)

# Pairwise comparison (11 metrics including deep chord similarity)
pair = pair_eval("generated.mid", "reference.mid")
print(pair.note_f1, pair.sim_chr, pair.ca, pair.cs)

# CLI
smg-eval -m generated.mid
smg-eval -p gen.mid -r ref.mid
smg-eval -m gen.mid -p gen.mid -r ref.mid -d -a -S -R
smg-eval -m gen.mid --only pce ebr gs --json

Installation

pip install smg-metrics

# Optional: Install torch for deep chord similarity (CS metric)
pip install smg-metrics[torch]
# Or: pip install torch>=2.0.0

# Or install from source:
git clone https://github.com/OlyMarco/smg_metric.git
cd smg_metric && pip install -e .

Package	Version	Purpose
muspy	>= 0.5.0	13 single-file quality metrics
miditoolkit	>= 1.0	MIDI parsing
pretty-midi	>= 0.2.10	Beat tracking & bar-level parsing
mir-eval	>= 0.7	Note-overlap metric
torch	>= 2.0.0	Optional: Deep chord similarity (CS metric)
numpy	>= 1.24	Numerical computation
scipy	>= 1.10	Scientific computing

Python API

from smg_metrics import (
    single_file,                # 13 MusPy quality metrics
    single_file_structural,     # 2 structural single-file metrics
    single_file_rhythmic,       # 4 D3PIA-style rhythmic metrics
    pair_eval,                  # 11 core pairwise metrics (incl. CS)
    pair_eval_structural,       # 2 structural pairwise metrics
    distribution_eval,          # 5 distribution-level metrics
    advanced_eval,              # 14 advanced metrics
    compute_ook,                # Out-of-Key percentage (FGG 2025)
    compute_cs,                 # Chord Similarity (Wang et al. 2020)
)

Container	Fields	Count
SingleFileResult	pce, ebr, gs, sc, pisr, polyphony, polyphony_rate, pitch_range, n_pitches_used, n_pitch_classes_used, emr, pe, dpc	13
StructuralSingleResult	che, ngram_div	2
RhythmicResult	mean_ioi, rhythmic_intensity, rhythmic_density, voice_number	4
PairResult	note_f1, notei_f1, mel_f1, i_iou, ver, sim_chr, sim_grv, ca, cs, onset_xor, note_overlap	11
StructuralPairResult	melody_match, tonal_dist	2
DistributionResult	pd, dd, sc_sim, pce_sim, gs_sim	5
AdvancedResult	kl_duration, kl_ioi, kl_pitch, oa_duration, oa_ioi, oa_pitch_range, oa_density, ci_precision, ci_recall, ci_f1, cts, cr_pred, cr_ref, recon_acc	14

Every result container is a frozen dataclass with .to_dict().

Chord Recognition

from smg_metrics import recognize_chords, compute_ca, compute_cs

# Advanced DP-based chord recognition (17 qualities + inversions)
chords = recognize_chords("music.mid")
for interval in chords:
    print(f"{interval.start:.2f}s - {interval.end:.2f}s: {interval.label}")
# Output: 0.00s - 2.50s: C:maj
#         2.50s - 5.00s: G:7/5  (second inversion)

# Chord Accuracy (rule-based DP method, FGG 2025)
ca = compute_ca("generated.mid", "reference.mid")
print(f"Chord Accuracy: {ca:.2%}")

# Chord Similarity (deep embedding, Wang et al. 2020)
cs = compute_cs("generated.mid", "reference.mid")
print(f"Chord Similarity: {cs:.4f}")

Note: CS metric requires downloading model weights (29 MB). See Model Weights section below.

Out-of-Key Notes

from smg_metrics import compute_ook

# Compute percentage of 16th-note steps with out-of-key notes
ook = compute_ook("generated.mid")
print(f"Out-of-Key: {ook:.2f}%")

# Get detailed breakdown
ook, details = compute_ook("generated.mid", return_details=True)
print(f"Key: {details['key']}")
print(f"OOK steps: {details['ook_steps']}/{details['total_steps']}")
print(f"OOK notes: {details['ook_notes']}")

Reference: FGG (Zhu et al., ICML 2025) uses OOK to measure dissonance. Well-controlled generation should have OOK ≈ 0–2%.

Model Weights

The Chord Similarity (CS) metric requires pretrained model weights:

Quick Download

# Download lightweight model (29 MB, recommended)
cd smg_metrics/model_weights
wget https://github.com/OlyMarco/smg_metric/releases/download/v5.1-models/polydis-v1-chd_encoder_only.pt

Model Details

• Architecture: Bidirectional GRU chord encoder (36 → 1024 → 256)
• Training: EC2-VAE (Wang et al., ISMIR 2020)
• Size: 29 MB (pruned from 104 MB full model, 72.3% reduction)
• License: Inherits from original PolyDisVAE

Citation:

@inproceedings{wang2020learning,
  title={Learning interpretable representation for controllable polyphonic music generation},
  author={Wang, Ziyu and Wang, Dingsu and Zhang, Yixiao and Xia, Gus},
  booktitle={Proceedings of the 21st International Society for Music Information Retrieval Conference},
  year={2020}
}

See smg_metrics/model_weights/README.md for more details.

Chord Recognition

from smg_metrics import recognize_chords, compute_ca

# Beat-level DP chord recognition (music-x-lab algorithm)
chords = recognize_chords("song.mid")
for iv in chords:
    print(f"{iv.start:.2f}-{iv.end:.2f}: {iv.label}")

# Chord Accuracy with DP method (default)
ca = compute_ca("pred.mid", "ref.mid", method="dp")

# Or use Viterbi method (GETMusic)
ca = compute_ca("pred.mid", "ref.mid", method="viterbi")

Individual metrics

from smg_metrics import (
    chord_histogram_entropy, ngram_diversity,
    melody_matchness, tonal_distance,
    compute_ca, midi_to_chords, midi_to_chords_dp,
    mean_ioi, rhythmic_intensity, rhythmic_density,
    voice_number, onset_xor_distance, note_overlap,
    grooving_pattern_similarity,
)

che = chord_histogram_entropy("file.mid")
ca = compute_ca("pred.mid", "ref.mid")
gs = grooving_pattern_similarity("pred.mid", "ref.mid")

CLI Usage

# Single-file quality (14 metrics: 13 MusPy + OOK)
smg-eval -m generated.mid

# Single-file + structural + rhythmic (20 metrics)
smg-eval -m generated.mid -S -R

# Pairwise core (11 metrics: includes CS with deep embedding)
smg-eval -p gen.mid -r ref.mid

# Full 53-metric run
smg-eval -m gen.mid -p gen.mid -r ref.mid -d -a -S -R

# Select specific metrics
smg-eval -m gen.mid --only pce ebr gs ook
smg-eval -p gen.mid -r ref.mid --only ca cs note_f1

# List all available metrics
smg-eval --list-metrics

# JSON output
smg-eval -m gen.mid --json

# Batch directory
smg-eval --pred_dir ./pred/ --ref_dir ./ref/

# Timing
smg-eval -m gen.mid --time

Flag	Description	Default
-m, --music PATH	Single-file mode	--
-p, --pred PATH	Predicted MIDI for pair mode	--
-r, --ref PATH	Reference MIDI for pair mode	--
--pred_dir DIR	Batch predicted directory	--
--ref_dir DIR	Batch reference directory	--
--root INT	Root pitch for PISR	0
--mode {major,minor}	Scale mode for PISR	major
-d, --dist	Distribution-level metrics	false
-a, --advanced	Advanced metrics	false
-S, --structural	Structural metrics	false
-R, --rhythmic	Rhythmic metrics	false
--only METRIC ...	Select specific metrics	--
--list-metrics	List all metric names	--
--json	JSON output	false
--time	Print elapsed time	false

Metrics Reference

A. Single-file Quality (14)

Sources: MusPy / ISMIR 2020, XMusic / IEEE 2025, FGG ICML 2025.

Metric	Symbol	Range	Reference
Pitch Class Entropy	PCE	[0, log₂12]	Wu & Yang, ISMIR 2020; XMusic 2025
Empty Beat Rate	EBR	[0, 1]	Dong et al., ISMIR 2018; XMusic 2025
Groove Consistency	GS	[0, 1]	Wu & Yang, ISMIR 2020; XMusic 2025
Scale Consistency	SC	[0, 1]	Mogren, NeurIPS-W 2016
Pitch-in-Scale Rate	PISR	[0, 1]	Dong et al., AAAI 2018
Polyphony	Poly	[1, ∞)	Dong et al., AAAI 2018
Polyphony Rate	PR	[0, 1]	Dong et al., AAAI 2018
Pitch Range	Range	[0, 127]	MusPy 2020
Unique Pitches	N_p	[0, 128]	MusPy 2020
Unique Pitch Classes	N_pc	[0, 12]	MusPy 2020
Empty Measure Rate	EMR	[0, 1]	Dong et al., AAAI 2018
Pitch Entropy	PE	[0, 7]	MusPy 2020
Drum Pattern Consistency	DPC	[0, 1]	Dong et al., AAAI 2018
Out-of-Key Percentage	OOK	[0, 100]	FGG 2025, Krumhansl-Kessler key detection

B. Note-level Pairwise (5)

Source: Ou et al., NeurIPS 2025.

Metric	Symbol	Range
Note F1	F1	[0, 1]
Notei F1	F1i	[0, 1]
Melody F1	F1mel	[0, 1]
Instrument IoU	I-IoU	[0, 1]
Voice Error Rate	VER	[0, ∞)

B2. Pairwise Rhythmic (3)

Sources: D3PIA ICASSP 2026, mir_eval ISMIR 2014, Wu & Yang ISMIR 2020.

Metric	Symbol	Range
Onset XOR Distance	XOR	[0, 1]
Note Overlap	NOvlp	[0, 1]
Grooving Pattern Similarity	GS_d3pia	[0, ∞)

C. Bar-level Pairwise (2)

Source: MuseMorphose, IEEE/ACM TASLP 2023.

Metric	Symbol	Range
Chroma Similarity	simChr	[0, 1]
Groove Similarity	simGrv	[0, 1]

D. Chord-level Pairwise (2)

Sources: FGG ICML 2025, Wang et al. ISMIR 2020, music-x-lab/midi-chord-recognition.

Metric	Symbol	Range	Description
Chord Accuracy	CA	[0, 1]	Beat-level DP chord recognition + exact match
Chord Similarity	CS	[0, 1]	Deep chord embedding similarity (requires PyTorch)

Chord Recognition Pipeline (adapted from music-x-lab):

1. Extract beat/downbeat positions from MIDI tempo map
2. Quantise notes to beat grid → per-beat 12-dim treble chroma + bass chroma
3. Channel-weighted aggregation (thickness + bass reweighting)
4. Score each chord template per beat (with bass bonus)
5. Dynamic-programming decode with span-length reward and transition penalty
6. Output interval-level chord labels

Two methods available: 'dp' (default, beat-level) and 'viterbi' (bar-level HMM).

E. Distribution-level (5)

Sources: SongMASS ACM-MM 2020, MusPy ISMIR 2020.

Metric	Symbol	Range
Pitch Distribution	PD	[0, 1]
Duration Distribution	DD	[0, 1]
Scale Consistency Sim	SC_sim	[0, 1]
Pitch Class Entropy Sim	PCE_sim	[0, 1]
Groove Consistency Sim	GS_sim	[0, 1]

F. Advanced (14)

Sources: GETMusic IJCAI 2025, Rule Guided ICML 2024, Text2midi AAAI 2025, MuseTok ICASSP 2026.

Metric	Symbol	Range
KL Divergence (Duration)	KL_dur	[0, ∞)
KL Divergence (IOI)	KL_ioi	[0, ∞)
KL Divergence (Pitch)	KL_pitch	[0, ∞)
Overlapping Area ×4	OA	[0, 1]
Instrument Coverage ×3	CI	[0, 1]
Correct Time Signature	CTS	{0, 1}
Compression Ratio ×2	CR	[0, ∞)
Reconstruction Accuracy	ReconAcc	[0, 1]

G. Structural (4)

Sources: Papadopoulos & Peeters ISMIR 2012, Yang & Lerch NCA 2018, Mongeau & Sankoff CH 1990, Harte et al. ACM MM 2006.

Metric	Symbol	Type	Range
Chord Histogram Entropy	CHE	single	[0, log₂C]
N-gram Diversity	Ngram	single	[0, 1]
Melody Matchness	MM	pair	[0, 1]
Tonal Distance	TD	pair	[0, ∞)

H. Rhythmic/Temporal Single-file (4)

Source: D3PIA ICASSP 2026, MIDISym feature extraction.

Metric	Symbol	Range
Mean IOI	IOI	[0, ∞)
Rhythmic Intensity	RI	[0, ∞)
Rhythmic Density	RD	[0, 1]
Voice Number	VN	[0, ∞)

v5.1 Changelog

New features

• Chord Similarity (CS): Deep chord embedding metric using pruned PolyDisVAE encoder (Wang et al. ISMIR 2020)
  • Optional dependency: requires torch>=2.0.0 (install with pip install smg-metrics[torch])
  • Lightweight model (29 MB): polydis-v1-chd_encoder_only.pt
  • Supports 17 chord qualities + inversions via dynamic programming recognition
• Out-of-Key (OOK): Percentage of notes outside detected key using Krumhansl-Kessler algorithm
  • Standalone single-file metric, no external annotations needed
  • Integrated into CLI with --only ook support

Changed

• PyTorch: Moved to optional dependencies (torch extra)
• Metric count: 51 → 53 (added CS + OOK)
• Version: 5.0.0 → 5.1.0

License

MIT — see LICENCE.

Citation

If you use smg-metrics in your research, please cite:

@software{smg_metrics,
  title  = {smg-metrics: Objective Evaluation Metrics for Symbolic Music Generation},
  author = {Temmie Pratt},
  year   = {2026},
  url    = {https://github.com/OlyMarco/smg_metric},
}