[Fix] make arguments effective in tools/confusion_matrix.py (open-mml…

…ab#1401)

* add an argument for customizing `title' of the output figure

* fix `color_theme' arguments not passing to plot function

Signed-off-by: code14 <mob5566@gmail.com>

tools/confusion_matrix.py

@@ -25,6 +25,10 @@ def parse_args():
         '--color-theme',
         default='winter',
         help='theme of the matrix color map')
+    parser.add_argument(
+        '--title',
+        default='Normalized Confusion Matrix',
+        help='title of the matrix color map')
     parser.add_argument(
         '--cfg-options',
         nargs='+',
@@ -171,7 +175,9 @@ def main():
         confusion_matrix,
         dataset.CLASSES,
         save_dir=args.save_dir,
-        show=args.show)
+        show=args.show,
+        title=args.title,
+        color_theme=args.color_theme)
 
 
 if __name__ == '__main__':