From c872e4bfa14b63eefc7c418d27ade963b678da69 Mon Sep 17 00:00:00 2001 From: Brian DuSell Date: Fri, 4 Dec 2020 16:00:11 -0600 Subject: [PATCH] Minor fix to plotting script. --- src/plot_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plot_train.py b/src/plot_train.py index 32c821d..cf651a5 100644 --- a/src/plot_train.py +++ b/src/plot_train.py @@ -146,7 +146,7 @@ def main(): parser.add_argument('--labels', nargs='+', default=[], help='Labels for each model used in the plot. There should be as many ' 'labels as models.') - parser.add_argument('--metric', default='perplexity-diff', + parser.add_argument('--metric', default='cross-entropy-diff', help='Which metric to plot on the y-axis. Default is difference in ' 'cross entropy between the model and the lower bound on the ' 'validation set.')