setRNNTorch: Create setting for RNN model with python

In hxia/plp-git-demo: Package for patient level prediction using data in the OMOP Common Data Model

Description

Create setting for RNN model with python

Usage

setRNNTorch(
  hidden_size = c(50, 100),
  epochs = c(20, 50),
  seed = 0,
  class_weight = 0,
  type = "RNN"
)

Arguments

hidden_size	The hidden size
epochs	The number of epochs
seed	A seed for the model
class_weight	The class weight used for imbalanced data: 0: Inverse ratio between positives and negatives -1: Focal loss
type	It can be normal 'RNN', 'BiRNN' (bidirectional RNN) and 'GRU'

Examples

## Not run: 
model.rnnTorch <- setRNNTorch()

## End(Not run)

hxia/plp-git-demo documentation built on March 19, 2021, 1:54 a.m.