R/HoFM.train.R

In FactoRizationMachines: Machine Learning with Higher-Order Factorization Machines

HoFM.train <-
function(data, target, factors=c(1,10,5), intercept=T, iter=100, regular=NULL, stdev=0.1){
  
  object=list()
  object$vK=factors
  
  # if(length(factors)>3) warning("HoFM.train only supports up to third-order factors -> parameter factors partly ignored")
  if(length(factors)>10) warning("HoFM.train only supports up to tenth-order factors -> parameter factors partly ignored")
  
  return(learn.FM.model(data=data, target=target, object=object, intercept=intercept, iter=iter, regular=regular, stdev=stdev))
  
}