def get_ks_dist()

in ai-judge/inference.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def get_ks_dist(midis, bach_pdf):
    '''Calculate KS similarity'''
    ks = []
    for midi in midis:
        midi = midi[0, :, :]
        density = pch(midi)
        d, p = stats.ks_2samp(density, bach_pdf)
        ks.append(d)
    return ks