Skip to content

Commit edb7380

Browse files
committed
correct typo
1 parent 0de12a6 commit edb7380

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def parseCommand():
2323
parser.add_option("-m", "--model", dest = "model_file", default = "cfdna.model",
2424
help = "specify which file to store the built model.")
2525
parser.add_option("-a", "--algorithm", dest = "algorithm", default = "knn",
26-
help = "specify which algorithm to use for classfication, candidates are svm/knn/rbf/rf/gnb/benchmark, rbf means svm using rbf kernel, rf means random forrest, gnb means Gaussian Naive Bayes, benchmark will try every algorithm and plot the score figure, default is knn.")
26+
help = "specify which algorithm to use for classfication, candidates are svm/knn/rbf/rf/gnb/benchmark, rbf means svm using rbf kernel, rf means random forest, gnb means Gaussian Naive Bayes, benchmark will try every algorithm and plot the score figure, default is knn.")
2727
parser.add_option("-c", "--cfdna_flag", dest = "cfdna_flag", default = "cfdna",
2828
help = "specify the filename flag of cfdna files, separated by semicolon. default is: cfdna")
2929
parser.add_option("-o", "--other_flag", dest = "other_flag", default = "gdna;ffpe",
@@ -226,7 +226,7 @@ def main():
226226
train(model, data, label, samples, options)
227227
elif options.algorithm.lower() == "benchmark":
228228
print("\nstarting benchmark...")
229-
names = ["KNN", "Random Forrest","SVM Linear", "Gaussian Naive Bayes", "SVM RBF"]
229+
names = ["KNN", "Random Forest","SVM Linear Kernel", "Gaussian Naive Bayes", "SVM RBF Kernel"]
230230
models = [neighbors.KNeighborsClassifier(leaf_size=100), RandomForestClassifier(n_estimators=20), svm.LinearSVC(), GaussianNB(), svm.SVC(kernel='rbf')]
231231
scores_arr = [train(model, data, label, samples, options, True) for model in models]
232232
print("ploting benchmark result...")

0 commit comments

Comments
 (0)