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