Carlos-Francisco Méndez-Cruz

Iris dataset for automatic clasification

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