Showing
1 changed file
with
1 additions
and
0 deletions
... | @@ -140,6 +140,7 @@ if __name__ == "__main__": | ... | @@ -140,6 +140,7 @@ 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 | + if options.classifier in ["Perceptron", "SVM"]: | ||
143 | confidence_scores = classifier.decision_function(dataEvaluation) | 144 | confidence_scores = classifier.decision_function(dataEvaluation) |
144 | print(" Training and predition done!") | 145 | print(" Training and predition done!") |
145 | 146 | ... | ... |
-
Please register or login to post a comment