train: Train model using BaTFLED algorthm

In nathanlazar/BaTFLED3D: Bayesian Tensor Factorization Linked to External Data

Description

Model objects are updated in place to avoid memory issues. Nothing is returned.

Usage

train(d, m, ...)

Arguments

d	an input data object created with input_data
m	a CP_model or Tucker_model object created with mk_model
...	extra arguments (params) passed to train_CP or train_Tucker

Examples

data.params <- get_data_params(c('decomp=Tucker'))
toy <- mk_toy(data.params)
train.data <- input_data$new(mode1.X=toy$mode1.X[,-1],
                             mode2.X=toy$mode2.X[,-1],
                             mode3.X=toy$mode3.X[,-1],
                             resp=toy$resp)
model.params <- get_model_params(c('decomp=Tucker'))
toy.model <- mk_model(train.data, model.params)
toy.model$rand_init(model.params)

train(d=train.data, m=toy.model, new.iter=1, params=model.params)

nathanlazar/BaTFLED3D documentation built on May 23, 2019, 12:19 p.m.