R/print.R

In juscelino-izidoro/supcavs: Supervised and Unsupervised PCA Variable Selection

print <- 
function(x, ...){
  UseMethod('print')
}

print.VarSel <-
function(x, ...){
  cat(paste(" Methods:", x$main, "(No. of registers = ", nrow(x$x), ")\n\n"))
  xnames = dimnames(x$x)[[2]]
  n = length(x$var)
  RemovedVariables = vector('integer', n)
  SelectedVariables   = vector('integer', n)
  RemovedVariablesNames = c()
  SelectedVariablesNames   = c()
  for(i in 1:n){ RemovedVariables[i] = length(x$var[[i]]) }
  SelectedVariables = ncol(x$x)-RemovedVariables
  
  for(i in 1:n){ 
    RemovedVariablesNames = c(RemovedVariablesNames,paste(x$var[[i]], collapse=',')) ;
    SelectedVariablesNames   = c(SelectedVariablesNames, paste(setdiff(xnames, x$var[[i]]), collapse=',')) ;
  }
  
  if(exists('super', where=x)){
    Super = vector('integer', n)
    
    for(i in 1:n){ Super[i] = x$super[i] } 
    
    print(data.frame(NumVarRem=RemovedVariables, 
                     VarsRems=RemovedVariablesNames,
                     Superv=Super, 
                     NumVarSel=SelectedVariables,
                     VarsSel=SelectedVariablesNames
                    ),
          row.names=F
    )
  }
  else{
    print(data.frame(NumVarRem=RemovedVariables, 
                     VarsRems=RemovedVariablesNames, 
                     NumVarSek=SelectedVariables,
                     VarsSel=SelectedVariablesNames
                    ), 
          row.names=F
    )
  }
  cat("\n")
  invisible(x)
}

print.VarSelPerformance <-
function(x, ...){
  cat(paste(" Methods:", x$main, "(No. of registers = ", x$nRegs, ")\n\n"))
  RemovedVariables  = c(0,x$x)
  SelectedVariables = c(x$nVars-RemovedVariables)
  Performance       = c(x$y_0,x$y)
  if(exists('super', where=x)){
    RemovedSupervision = rbind(0,x$super)
    print(data.frame(Removed=RemovedVariables, Selected=SelectedVariables, Supervision=RemovedSupervision, Performance=Performance), row.names=F)
  }
  else{
    print(data.frame(Removed=RemovedVariables, Selected=SelectedVariables,  Performance=Performance), row.names=F)
  }
  cat("\n")
  invisible(x)
}