Create setting for CNN model with python
setCNNTorch( nbfilters = c(16, 32), epochs = c(20, 50), seed = 0, class_weight = 0, type = "CNN" )
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' |
if (FALSE) { model.cnnTorch <- setCNNTorch() }