def evaluate()

in source/sagemaker/src/package/data_privatization/container/train.py [0:0]

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

def evaluate(model, iterator, criterion):
    epoch_loss = 0
    epoch_acc = 0

    model.eval()

    for batch in iterator:
        predictions = model(batch.review).squeeze(1)

        loss = criterion(predictions, batch.sentiment)

        acc = binary_accuracy(predictions, batch.sentiment)

        epoch_loss += loss.item()
        epoch_acc += acc.item()

    return epoch_loss / iterator.iterations, epoch_acc / iterator.iterations