Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing binding thrombin dataset

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