Fix minor naming issue in embedding model 2

code_mario/clef18_task1_emb2.py

@@ -583,7 +583,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser(prog="CLEF2018")
     subparsers = parser.add_subparsers(dest="mode")
 
-    train_emb_parser = subparsers.add_parser("train-emb2")
+    train_emb_parser = subparsers.add_parser("train-emb")
     train_emb_parser.add_argument("lang", help="Language to train on", choices=["it", "fr", "hu"])
     train_emb_parser.add_argument("--epochs", help="Number of epochs to train", default=10, type=int)
     train_emb_parser.add_argument("--batch_size", help="Batch size during training", default=10, type=int)
@@ -634,7 +634,7 @@ if __name__ == "__main__":
     clef18_task1 = Clef18Task1Emb2()
     clef18_task1.save_arguments(args)
 
-    if args.mode == "train-emb2":
+    if args.mode == "train-emb":
         configuration = clef18_task1.prepare_data_set(certificates, dictionary, ft_model, args.train_ratio, args.val_ratio,args.strat_column,
                                                       args.samples, args.strat_splits)
         clef18_task1.save_configuration(configuration)