INNER CODE UNIT · Python

compute_metrics

microsoft/muzic · musecoco/1-text2attribute_model/main.py:382

    def compute_metrics(p: EvalPrediction):
        preds = p.predictions
        labels = p.label_ids
        acc = {}
        for k,v in preds.items():
            pred = np.argmax(preds[k], axis=1)
            acc[k] = accuracy_score(labels[k], pred)

        return {"acc": acc}

    # Data collator will default to DataCollatorWithPadding when the tokenizer is passed to Trainer, so we change it if
    # we already did the padding.
    if data_args.pad_to_max_length:
        data_collator = default_data_collator
    elif training_args.fp16:
        data_collator = DataCollatorWithPadding(tokenizer, pad_to_multiple_of=8)
    else:
        data_collator = None

View source record →