Skip to contents

Fit a GEM model

Usage

fit_model(
  x,
  y,
  sample_map,
  lr = 0.1,
  iters = 180,
  num_threads = 4L,
  size = NULL,
  verbose = FALSE
)

Arguments

x

A numeric matrix of predictors

y

A numeric response vector

sample_map

A list of sample indices

lr

Learning rate

iters

Number of iterations

num_threads

Number of threads to set

size

Optional subsample size

Value

A list containing model parameters and fitted values