find_best_neural_net: Does a search for the best neural net to fit the data using a...

Description Usage Arguments Value

Description

Does a search for the best neural net to fit the data using a restricted h2o.automl function. See this function in h2o documentation for more info. You will need the h2o package. And you may have to download an older version of java for it to work.

Usage

1
2
find_best_neural_net(Embeddings, dependent_var, max_minutes = 10,
  stopping_tol = 0.005, n_folds = 5)

Arguments

Embeddings

The embedding dataframe or matrix

dependent_var

The variable used to predict, must be a factor

max_minutes

The maximum run time for model search if stopping tolerance not reached. For larger data sets you may need multiple hours. Defaults to 10

stopping_tol

The stopping tolerance for model search. Defaults to .005

nfolds

The number of folds for k-fold cross-validation. Defaults to 5

Value

Returns the best model


adrianapaza/easynlp documentation built on May 9, 2019, 7:31 p.m.