Carlos-Francisco Méndez-Cruz

Iris dataset for automatic clasification

......@@ -162,8 +162,9 @@ 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":
oFile.write("{}".format(confidence_scores))
if options.classifier in ["Perceptron", "SVM"]:
oFile.write('Confidence scores: \n')
oFile.write("{}".format(confidence_scores))
print(" Saving test report done!")
......