Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing binding thrombin dataset

...@@ -28,7 +28,7 @@ __author__ = 'CMendezC' ...@@ -28,7 +28,7 @@ __author__ = 'CMendezC'
28 # 6) --outputModelFile File to place output model. 28 # 6) --outputModelFile File to place output model.
29 # 7) --outputReportPath Path to place evaluation report. 29 # 7) --outputReportPath Path to place evaluation report.
30 # 8) --outputReportFile File to place evaluation report. 30 # 8) --outputReportFile File to place evaluation report.
31 -# 9) --classifier Classifier: BernoulliNB, SVM, NearestCentroid. 31 +# 9) --classifier Classifier: BernoulliNB, SVM, kNN.
32 # 10) --saveData Save matrices 32 # 10) --saveData Save matrices
33 # 11) --kernel Kernel 33 # 11) --kernel Kernel
34 # 12) --reduction Feature selection or dimensionality reduction 34 # 12) --reduction Feature selection or dimensionality reduction
...@@ -80,7 +80,7 @@ if __name__ == "__main__": ...@@ -80,7 +80,7 @@ if __name__ == "__main__":
80 help="File to place evaluation report", metavar="FILE") 80 help="File to place evaluation report", metavar="FILE")
81 parser.add_argument("--classifier", dest="classifier", 81 parser.add_argument("--classifier", dest="classifier",
82 help="Classifier", metavar="NAME", 82 help="Classifier", metavar="NAME",
83 - choices=('BernoulliNB', 'SVM', 'NearestCentroid'), default='SVM') 83 + choices=('BernoulliNB', 'SVM', 'kNN'), default='SVM')
84 parser.add_argument("--saveData", dest="saveData", action='store_true', 84 parser.add_argument("--saveData", dest="saveData", action='store_true',
85 help="Save matrices") 85 help="Save matrices")
86 parser.add_argument("--kernel", dest="kernel", 86 parser.add_argument("--kernel", dest="kernel",
......