diff --git a/src/victim/train.py b/src/victim/train.py index f335444..30a5d4c 100644 --- a/src/victim/train.py +++ b/src/victim/train.py @@ -60,6 +60,10 @@ ) if args.evaluate: - print(classifier.evaluate()) + result = classifier.evaluate() + print(f"[*] Evaluation of {args.dataset.upper()} Classifier") + for key in result: + print(f" - {key}: {result[key]}") + else: classifier.train(e)