changeset 3:a5a5716e0317 draft

Uploaded
author testtool
date Fri, 13 Oct 2017 10:14:29 -0400
parents 6169ba9ed42a
children 4494c973f643
files accuracy.R
diffstat 1 files changed, 48 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/accuracy.R	Fri Oct 13 10:14:29 2017 -0400
@@ -0,0 +1,48 @@
+require(caret, quietly = TRUE)
+
+args <- commandArgs(trailingOnly = TRUE)
+
+input = args[1]
+p = args[2]   
+output1 = args[3] 
+output2 = args[4] 
+
+dataset <- read.csv(input, header=TRUE)
+
+validation_index <- createDataPartition(dataset$Species, p=p, list=FALSE)
+
+validation <- dataset[-validation_index,]
+
+validdataset <- dataset[validation_index,]
+
+percentage <- prop.table(table(validdataset$Species)) * 100
+cbind(freq=table(validdataset$Species), percentage=percentage)
+
+output_summary <- summary(validdataset) 
+write.csv(output_summary,output1)
+
+control <- trainControl(method="cv", number=10)
+metric <- "Accuracy"
+
+# a) linear algorithms
+set.seed(7)
+fit.lda <- train(Species~., data=validdataset, method="lda", metric=metric, trControl=control)
+# b) nonlinear algorithms
+# CART
+set.seed(7)
+fit.cart <- train(Species~., data=validdataset, method="rpart", metric=metric, trControl=control)
+# kNN
+set.seed(7)
+fit.knn <- train(Species~., data=validdataset, method="knn", metric=metric, trControl=control)
+# c) advanced algorithms
+# SVM
+set.seed(7)
+fit.svm <- train(Species~., data=validdataset, method="svmRadial", metric=metric, trControl=control)
+# Random Forest
+set.seed(7)
+fit.rf <- train(Species~., data=validdataset, method="rf", metric=metric, trControl=control)
+
+results <- resamples(list(lda=fit.lda, cart=fit.cart, knn=fit.knn, svm=fit.svm, rf=fit.rf))
+output_results <- summary(results) 
+
+write.csv(as.matrix(output_results),output2)