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__":
print("File to read evaluation true classes: " + str(args.inputEvaluationClasses))
print("Path to place output files: " + str(args.outputPath))
print("File to write evaluation report: " + str(args.outputFile))
print("Classifier: " + str(args.outputFile))
print("Classifier: " + str(args.classifier))
# Start time
t0 = time()
......@@ -123,7 +123,7 @@ if __name__ == "__main__":
if args.classifier == "MultinomialNB":
classifier = MultinomialNB()
elif args.classifier == "SVM":
classifier = SVC()
classifier = SVC(kernel="linear")
elif args.classifier == "DecisionTree":
classifier = DecisionTreeClassifier()
elif args.classifier == "Perceptron":
......