setCNNTorch: Create setting for CNN 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 CNN model with python

Usage

setCNNTorch(
  nbfilters = c(16, 32),
  epochs = c(20, 50),
  seed = 0,
  class_weight = 0,
  type = "CNN"
)

Arguments

nbfilters	The number of filters
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 'CNN', 'CNN_LSTM', CNN_MLF' with multiple kernels with different kernel size, 'CNN_MIX', 'ResNet' and 'CNN_MULTI'

Examples

## Not run: 
model.cnnTorch <- setCNNTorch()

## End(Not run)

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