diff options
| author | John Bauer <horatio@gmail.com> | 2022-08-31 10:24:49 +0300 |
|---|---|---|
| committer | John Bauer <horatio@gmail.com> | 2022-08-31 10:24:49 +0300 |
| commit | 1169177e62acb17e362b864f914d58ee85ca5664 (patch) | |
| tree | 9f5402343747719b78be52c70b106ba0cb7833cf | |
| parent | 1e7ab92c8cf0f2f51f78df74472a6196b51f4637 (diff) | |
Oops, correct a few uses of model in the classifier main program
| -rw-r--r-- | stanza/models/classifier.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/stanza/models/classifier.py b/stanza/models/classifier.py index b580e0fc..137e3212 100644 --- a/stanza/models/classifier.py +++ b/stanza/models/classifier.py @@ -487,8 +487,8 @@ def main(args=None): save_name = args.save_name if not(save_name): save_name = args.base_name + "_" + args.shorthand + "_" - save_name = save_name + "FS_%s_" % "_".join([str(x) for x in model.config.filter_sizes]) - save_name = save_name + "C_%d_" % model.config.filter_channels + save_name = save_name + "FS_%s_" % "_".join([str(x) for x in args.filter_sizes]) + save_name = save_name + "C_%d_" % args.filter_channels if model.config.fc_shapes: save_name = save_name + "FC_%s_" % "_".join([str(x) for x in model.config.fc_shapes]) save_name = save_name + "classifier.pt" @@ -538,7 +538,7 @@ def main(args=None): test_set = data.read_dataset(args.test_file, args.wordvec_type, min_len=None) logger.info("Using test set: %s" % args.test_file) - data.check_labels(model.labels, test_set) + data.check_labels(trainer.model.labels, test_set) if args.test_remap_labels is None: confusion_matrix = confusion_dataset(trainer.model, test_set) |