机器学习 lasso回归 筛选关键基因

lasso_select = function(gene_exp,sample_info,num){
  
  data = t(gene_exp)
  data = bind_cols(sample_info,data)
  model = glmnet(x = data[,-1],
                 y = data[,1])
  
  model_new <- cv.glmnet(x = data[,-1] %>% as.matrix(),
                         y = data[,1],
                         nfolds = num)
  
  coe <- coef(model_new, s = model_new$lambda.min)
  act_index <- which(coe != 0)
  act_coe <- coe[act_index]
  lassogene=row.names(coe)[act_index]
  
  result =  data.frame(selected_gene_id=lassogene,coef=act_coe)
  result = result[-1,]

  return(result)

}