Showing
1 changed file
with
2 additions
and
1 deletions
... | @@ -140,7 +140,8 @@ if __name__ == "__main__": | ... | @@ -140,7 +140,8 @@ if __name__ == "__main__": |
140 | classifier.fit(dataTraining, trueTrainingClasses) | 140 | classifier.fit(dataTraining, trueTrainingClasses) |
141 | print(" Prediction...") | 141 | print(" Prediction...") |
142 | y_pred = classifier.predict(dataEvaluation) | 142 | y_pred = classifier.predict(dataEvaluation) |
143 | - confidence_scores = classifier.decision_function(dataEvaluation) | 143 | + if options.classifier in ["Perceptron", "SVM"]: |
144 | + confidence_scores = classifier.decision_function(dataEvaluation) | ||
144 | print(" Training and predition done!") | 145 | print(" Training and predition done!") |
145 | 146 | ||
146 | # for i in range(len(trueClasses)): | 147 | # for i in range(len(trueClasses)): | ... | ... |
-
Please register or login to post a comment