Carlos-Francisco Méndez-Cruz / laigen-supervised-learning

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
59e7490b 1 parent a3e4ae80

Classification binding thrombin data set

Inline Side-by-side
Showing 2 changed files with 6 additions and 4 deletions
...@@ -199,15 +199,17 @@ if __name__ == "__main__": ...@@ -199,15 +199,17 @@ if __name__ == "__main__":
199 classifier = SVC() 199 classifier = SVC()
200 if args.kernel == 'rbf': 200 if args.kernel == 'rbf':
201 paramGrid = {'C': scipy.stats.expon(scale=100), 201 paramGrid = {'C': scipy.stats.expon(scale=100),
202 - 'gamma': scipy.stats.expon(scale=.1), 202 + # 'gamma': scipy.stats.expon(scale=.1),
203 - 'kernel': ['rbf'], 'class_weight': ['balanced', None]} 203 + 'kernel': ['rbf'],
204 + 'class_weight': ['balanced', None]}
204 elif args.kernel == 'linear': 205 elif args.kernel == 'linear':
205 paramGrid = {'C': scipy.stats.expon(scale=100), 206 paramGrid = {'C': scipy.stats.expon(scale=100),
206 'kernel': ['linear'], 207 'kernel': ['linear'],
207 'class_weight': ['balanced', None]} 208 'class_weight': ['balanced', None]}
208 elif args.kernel == 'poly': 209 elif args.kernel == 'poly':
209 paramGrid = {'C': scipy.stats.expon(scale=100), 210 paramGrid = {'C': scipy.stats.expon(scale=100),
210 - 'gamma': scipy.stats.expon(scale=.1), 'degree': [2, 3], 211 + # 'gamma': scipy.stats.expon(scale=.1),
212 + 'degree': [2, 3],
211 'kernel': ['poly'], 'class_weight': ['balanced', None]} 213 'kernel': ['poly'], 'class_weight': ['balanced', None]}
212 myClassifier = model_selection.RandomizedSearchCV(classifier, 214 myClassifier = model_selection.RandomizedSearchCV(classifier,
213 paramGrid, n_iter=nIter, 215 paramGrid, n_iter=nIter,
......
...@@ -158,7 +158,7 @@ if __name__ == "__main__": ...@@ -158,7 +158,7 @@ if __name__ == "__main__":
158 if args.classifier == "BernoulliNB": 158 if args.classifier == "BernoulliNB":
159 classifier = BernoulliNB() 159 classifier = BernoulliNB()
160 elif args.classifier == "SVM": 160 elif args.classifier == "SVM":
161 - classifier = SVC() 161 + classifier = SVC(kernel="linear")
162 elif args.classifier == "kNN": 162 elif args.classifier == "kNN":
163 classifier = KNeighborsClassifier() 163 classifier = KNeighborsClassifier()
164 else: 164 else:
......