#!/bin/bash -x

# Exit on error
set -e

# Test code goes here
rm -rf perplexity perplexity.log
mkdir -p perplexity

$MRT_MARIAN/build/marian \
    --cost-type perplexity \
    --seed 9999 \
    -m perplexity/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \
    --disp-freq 2 --after-epochs 1 \
    --log perplexity.log

test -e perplexity/model.npz
test -e perplexity.log

cat perplexity.log | grep 'Ep\. 1 :' | $MRT_TOOLS/extract-costs.sh > perplexity.out
$MRT_TOOLS/diff-floats.py perplexity.out perplexity.expected -p 0.5 > perplexity.diff

# Exit with success code
exit 0