--- language: en tags: - sagemaker - bert-base-uncased - text classification license: apache-2.0 datasets: - adecorpusv2 model-index: - name: bert-base-uncased results: - task: name: Text Classification type: text-classification dataset: name: "ade_corpus_v2Ade_corpus_v2_classification" type: ade_corpus metrics: -name: Validation accuracy type: accuracy value: 0.9298021697511167 -name: Validation f1 type: f1 value: 0.827315541601256 -name: eValidation loss type: loss value: 0.17835010588169098 -name: Validation recall type: recall value: 0.823437 -name: Validation precision type: precision value: 0.831230283911672 --- ## bert-base-uncased This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container. ## Hyperparameters ```json { "do_eval": true, "do_train": true, "fp16": true, "load_best_model_at_end": true, "model_name": "bert-base-uncased", "num_train_epochs": 10, "per_device_eval_batch_size": 16, "per_device_train_batch_size": 16, "learning_rate":5e-5 } ``` ## Validation Metrics | key | value | | --- | ----- | | eval_accuracy | 0.9298021697511167 | | eval_auc | 0.8902672664394546 | | eval_f1 | 0.827315541601256 | | eval_loss | 0.17835010588169098 | | eval_recall | 0.8234375 | | eval_precision | 0.831230283911672 | 9cec7af Jorge Lopez Grisman commited on Nov 22, 2021