R/network_classify.R

In xinying17/FeatureV3:

# main function
network_classify <- function(L='label',data_train,data_test,nf,p,corr,f_type,s,nc,kern){

  nc = round(max(1,min(nc,nf/5)))
  newdata <- network_features(L='label',data_train,data_test,nf,p,corr,f_type,s,nc)

  # test
  require('e1071', quietly = TRUE)
  library(e1071)

  wts <- nrow(newdata$new_train) / table(newdata$train_label)
  model1 <- svm(newdata$new_train,newdata$train_label,type="C-classification",class.weights = wts,kernel = kern)
  prediction <- predict(model1, newdata$new_test)

  return(list(pred = prediction, acc = sum(prediction==newdata$test_label)/nrow(data_test)))
}