Learn R Programming

PatientLevelPrediction (version 3.0.0)

setRNNTorch: Create setting for RNN model with python

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

Run this code
# NOT RUN {
model.rnnTorch <- setRNNTorch()
# }

Run the code above in your browser using DataLab