From 9568b6e2e77cd86418a22eb7eabfda168b574c7a Mon Sep 17 00:00:00 2001 From: Jean-Baptiste Camps Date: Tue, 7 Jan 2025 15:21:37 +0100 Subject: [PATCH] forgot --- load_corpus.py | 8 ++++---- train_svm.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/load_corpus.py b/load_corpus.py index 545f0f2a..80a6ac75 100755 --- a/load_corpus.py +++ b/load_corpus.py @@ -97,10 +97,10 @@ feat_file = "feature_list_{}{}grams{}mf.json".format(args.t, args.n, args.k) corpus_file = "feats_tests_n{}_k_{}.csv".format(args.n, args.k) - if not args.f: - with open(feat_file, "w") as out: - out.write(json.dumps(my_feats, ensure_ascii=False, indent=0)) - print("Features list saved to " + feat_file) + #if not args.f and : + with open(feat_file, "w") as out: + out.write(json.dumps(my_feats, ensure_ascii=False, indent=0)) + print("Features list saved to " + feat_file) corpus.to_csv(corpus_file) print("Corpus saved to " + corpus_file) diff --git a/train_svm.py b/train_svm.py index 119569de..615689e2 100755 --- a/train_svm.py +++ b/train_svm.py @@ -17,7 +17,7 @@ "If group_k-fold is chosen, each source file will be considered a group " "(only relevant if sampling was performed and more than one file per class was provided)", default=None, choices=['leave-one-out', 'k-fold', 'group-k-fold'], type=str) - parser.add_argument('--k', action='store', help="k for k-fold", default=10, type=int) + parser.add_argument('--k', action='store', help="k for k-fold (default: 10 folds for k-fold; k=n groups for group-k-fold)", default=0, type=int) parser.add_argument('--dim_reduc', action='store', choices=['pca'], help="optional dimensionality " "reduction of input data", default=None) parser.add_argument('--norms', action='store_true', help="perform normalisations? (default: True)",