pytorch-ner

pytorch-ner copied to clipboard

add schedulers

2

add all pytorch schedulers support

dayyass

dayyass

add: - classification report - confusion matrix

dayyass

add model checkpoint

1

Add model checkpoint This commit adds checkpoint.py file which allows: 1. Saving weights based on lowest val_loss from flag save_best_weights in config.yml 2. Saving weights frequently with specified frequency int...

abdallah1097

add early-stopping

1

Adding Early Stopping Feature This commits adds early stopping feature, early stopping class is created by Bjarten/early-stopping-pytorch [link below] Added several variables in config.yml: 1. early_stopping_flag: wether to enable early_stopping...

abdallah1097

add max norm (and other params) in embedding

dayyass

add config parametrization for: - full path to modules - neural_network architecture (BiLSTM, BiLSTMAttn, ...) - embedding (pre-trained or not) - embedding dropout (SpatialDropout, WordEmbeddingsDropout) - attention - ...

dayyass

MatanIvanov

dayyass

add config parametrization for: - full path to modules - neural_network architecture (BiLSTM, BiLSTMAttn, ...) - embedding (pre-trained or not) - embedding dropout (SpatialDropout, WordEmbeddingsDropout) - attention - ...

dayyass