fit_keras_es: Fit neural network by keras package with early stopping

In five-dots/ml4e: Machine Learning for Everyone

Description

Fit neural network by keras package with early stopping

Usage

fit_keras_es(
  x,
  y,
  eval_x,
  eval_y,
  task_type,
  keras_model,
  epoch = 10000L,
  patience = 20L,
  seed,
  ...
)

Arguments

x	A matrix of features.
y	A vector of labels.
eval_x	A matrix of features for evaluation.
eval_y	A vector of labels for evaluation.
task_type	A character scalar of task_type.
keras_model	A keras model object.
epoch	A integer scalar of epoch.
patience	A integer scalar of patience.
seed	A integer scalar of random seed.
...	Additional arguments passed to keras::fit.

Value

A fitted object.

five-dots/ml4e documentation built on June 19, 2020, 4:26 p.m.