set_model_params: Set model parameters.

Description Usage Arguments Value

View source: R/utils.R

Description

Set model parameters.

Usage

1
2
3
4
5
6
7
8
9
set_model_params(
  input_shape,
  num_classes,
  loss,
  activation,
  optimizer,
  metrics,
  ...
)

Arguments

input_shape

4D (x,y,z,channel) vector for input dimensions.

num_classes

Number of classes used for classification.

loss

Loss function (in quotes, e.g. "binary_crossentropy")

activation

Activation function (in quotes, e.g. "sigmoid")

optimizer

Optimizer as a keras function (e.g. keras::optimizer_adam())

metrics

Evaluation metrics; if more than 1, then make list.

...

Any additional parameters for specific models.

Value

List of model parameters.


willi3by/niiMLr documentation built on Dec. 23, 2021, 5:14 p.m.