Carlos-Francisco Méndez-Cruz

Classification Iris data set

......@@ -151,9 +151,9 @@ if __name__ == "__main__":
oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n')
if args.classifier in ["Perceptron", "SVM"]:
oFile.write('\nWeights assigned to the features: \n')
oFile.write("{}".format(classifier.coef_))
oFile.write('\nConfidence scores: \n')
oFile.write("{}".format(confidence_scores))
oFile.write("{}\n".format(classifier.coef_))
oFile.write('Confidence scores: \n')
oFile.write("{}\n".format(confidence_scores))
if args.classifier == "SVM":
oFile.write('Number of support vectors per class: \n{}\n'.format(classifier.n_support_))
oFile.write('Support vectors: \n{}\n'.format(classifier.support_vectors_))
......