Carlos-Francisco Méndez-Cruz

Iris dataset for automatic clasification

......@@ -162,7 +162,8 @@ if __name__ == "__main__":
oFile.write(str(confusion_matrix(trueEvaluationClasses, y_pred)) + '\n')
oFile.write('Classification report: \n')
oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n')
#if options.classifier == "Perceptron":
if options.classifier in ["Perceptron", "SVM"]:
oFile.write('Confidence scores: \n')
oFile.write("{}".format(confidence_scores))
print(" Saving test report done!")
......