RstPointerSegmenterTrainArgs

class RstPointerSegmenterTrainArgs(train_data_dir: str, test_data_dir: str, save_dir: str, hidden_dim: int = 64, rnn: str = 'GRU', num_rnn_layers: int = 6, use_bilstm: bool = True, lr: float = 0.01, dropout: float = 0.2, weight_decay: float = 0.0001, seed: int = 550, batch_size: int = 80, lr_decay_epoch: int = 10, use_batch_norm: bool = True, elmo_size: str = 'Large', epochs: int = 100)[source]