def get_class_score()

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

• 7 lines of code
• 1 McCabe index (conditional complexity)

def get_class_score(midis, model):
    '''Get classification score to bach music on selected midis'''
    sm = Softmax(dim=1)
    outputs = model(midis)
    probs = sm(outputs)
    scores = (probs[:,1] - probs[:,0] + 1) * 50
    return scores