Carlos-Francisco Méndez-Cruz

Iris dataset for automatic clasification

...@@ -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
......