Skip to contents

Estimate time to train a model

Source: R/ml_pipeline.R
morie_ml_train_time.Rd

Estimate time to train a model

Usage

morie_ml_train_time(model, x, y, probe_epochs = 5L)

Arguments

model

A morie_ml_model.

x

Predictors.

y

Response.

probe_epochs

Epochs to time for the estimate (default 5).

Value

Estimated seconds to run the full model$epochs.

Examples

morie_ml_train_time(morie_ml_model("linear", "gd", epochs = 100),
                    mtcars[c("hp", "wt")], mtcars$mpg)
#> [1] 0