Merge pull request #83 from SupervisedStylometry/culling

forgot

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)

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)",