Carlos-Francisco Méndez-Cruz / laigen-supervised-learning

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
aa4eedf9 1 parent a00f3bba

Classification Iris data set

Inline Side-by-side
Showing 1 changed file with 2 additions and 2 deletions
...@@ -72,7 +72,7 @@ if __name__ == "__main__": ...@@ -72,7 +72,7 @@ if __name__ == "__main__":
72 print("File to read evaluation true classes: " + str(args.inputEvaluationClasses)) 72 print("File to read evaluation true classes: " + str(args.inputEvaluationClasses))
73 print("Path to place output files: " + str(args.outputPath)) 73 print("Path to place output files: " + str(args.outputPath))
74 print("File to write evaluation report: " + str(args.outputFile)) 74 print("File to write evaluation report: " + str(args.outputFile))
75 - print("Classifier: " + str(args.outputFile)) 75 + print("Classifier: " + str(args.classifier))
76 76
77 # Start time 77 # Start time
78 t0 = time() 78 t0 = time()
...@@ -123,7 +123,7 @@ if __name__ == "__main__": ...@@ -123,7 +123,7 @@ if __name__ == "__main__":
123 if args.classifier == "MultinomialNB": 123 if args.classifier == "MultinomialNB":
124 classifier = MultinomialNB() 124 classifier = MultinomialNB()
125 elif args.classifier == "SVM": 125 elif args.classifier == "SVM":
126 - classifier = SVC() 126 + classifier = SVC(kernel="linear")
127 elif args.classifier == "DecisionTree": 127 elif args.classifier == "DecisionTree":
128 classifier = DecisionTreeClassifier() 128 classifier = DecisionTreeClassifier()
129 elif args.classifier == "Perceptron": 129 elif args.classifier == "Perceptron":
......