INNER CODE UNIT · Python

loss

microsoft/AzureML-BERT · finetune/run_classifier_azureml.py:138

        loss = tr_loss/nb_tr_steps if args.do_train else None

        result['eval_loss'] = eval_loss
        result['global_step'] = global_step
        result['loss'] = loss
        logger.info("***** Evaluation results *****")
        for key in sorted(result.keys()):
            logger.info("Epoch %s:  %s = %s", epoch_num,
                        key, str(result[key]))
            if(epoch_num  ==2):
                run.log(key, str(result[key]))
    if set_type == "test":
        output_eval_file = os.path.join(args.output_dir, f"{task_name.upper()}-{args.seed}-{args.learning_rate}-ep-{epoch_num}-tot-epochs-{args.num_train_epochs}.tsv")
        with open(output_eval_file, "w") as writer:
            writer.write("index\tprediction\n")
            for i, sample in enumerate(examples):
                if output_mode == "classification":
                    writer.write(f"{sample.guid}\t{processor.get_labels()[preds[i].item()]}\n")

View source record →