Carlos-Francisco Méndez-Cruz / lcg-bioinfoI-bionlp

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
951089e5 1 parent 1e93ca39

Iris dataset for automatic clasification

Inline Side-by-side
Showing 1 changed file with 3 additions and 2 deletions
...@@ -162,8 +162,9 @@ if __name__ == "__main__": ...@@ -162,8 +162,9 @@ if __name__ == "__main__":
162 oFile.write(str(confusion_matrix(trueEvaluationClasses, y_pred)) + '\n') 162 oFile.write(str(confusion_matrix(trueEvaluationClasses, y_pred)) + '\n')
163 oFile.write('Classification report: \n') 163 oFile.write('Classification report: \n')
164 oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n') 164 oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n')
165 - #if options.classifier == "Perceptron": 165 + if options.classifier in ["Perceptron", "SVM"]:
166 - oFile.write("{}".format(confidence_scores)) 166 + oFile.write('Confidence scores: \n')
167 + oFile.write("{}".format(confidence_scores))
167 168
168 print(" Saving test report done!") 169 print(" Saving test report done!")
169 170
......