Mercurial > hg > plosone_underreview
comparison scripts/util_feature_learning.py @ 10:8e897e82af51 branch-tests
edits in feature learning
| author | Maria Panteli <m.x.panteli@gmail.com> |
|---|---|
| date | Tue, 12 Sep 2017 13:31:42 +0100 |
| parents | c4841876a8ff |
| children | e8084526f7e5 |
comparison
equal
deleted
inserted
replaced
| 9:c4841876a8ff | 10:8e897e82af51 |
|---|---|
| 145 self.modelKNN = KNeighborsClassifier(n_neighbors=3, metric='euclidean') | 145 self.modelKNN = KNeighborsClassifier(n_neighbors=3, metric='euclidean') |
| 146 self.modelLDA = LDA() | 146 self.modelLDA = LDA() |
| 147 self.modelSVM = svm.SVC(kernel='rbf', gamma=0.1) | 147 self.modelSVM = svm.SVC(kernel='rbf', gamma=0.1) |
| 148 self.modelRF = RandomForestClassifier() | 148 self.modelRF = RandomForestClassifier() |
| 149 model_labels = ['KNN', 'LDA', 'SVM', 'RF'] | 149 model_labels = ['KNN', 'LDA', 'SVM', 'RF'] |
| 150 models = [modelKNN, modelLDA, modelSVM, modelRF] | 150 models = [self.modelKNN, self.modelLDA, self.modelSVM, self.modelRF] |
| 151 df_results = pd.DataFrame() | 151 df_results = pd.DataFrame() |
| 152 for model, model_label in zip(models, model_labels): | 152 for model, model_label in zip(models, model_labels): |
| 153 acc, _ = self.classification_accuracy(X_train, Y_train, X_test, Y_test, model=model) | 153 acc, _ = self.classification_accuracy(X_train, Y_train, X_test, Y_test, model=model) |
| 154 print model_label + " " + transform_label + " " + str(acc) | 154 print model_label + " " + transform_label + " " + str(acc) |
| 155 df_results = df_results.append(pd.DataFrame([[model_label, acc]])) | 155 df_results = df_results.append(pd.DataFrame([[model_label, acc]])) |