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
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