From 8744402f1eb51c7ae6b86cae1015983096beb655 Mon Sep 17 00:00:00 2001 From: Felix MIKAELIAN <39884124+fmikaelian@users.noreply.github.com> Date: Thu, 13 Feb 2020 23:19:13 +0100 Subject: [PATCH] add model_card flaubert-base-uncased-squad (#2833) * add model_card * Add tag cc @fmikaelian Co-authored-by: Julien Chaumond --- .../flaubert-base-uncased-squad/README.md | 48 +++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 model_cards/fmikaelian/flaubert-base-uncased-squad/README.md diff --git a/model_cards/fmikaelian/flaubert-base-uncased-squad/README.md b/model_cards/fmikaelian/flaubert-base-uncased-squad/README.md new file mode 100644 index 0000000000..fd7f5d9613 --- /dev/null +++ b/model_cards/fmikaelian/flaubert-base-uncased-squad/README.md @@ -0,0 +1,48 @@ +--- +language: french +--- + +# flaubert-base-uncased-squad + +## Description + +A baseline model for question-answering in french ([flaubert](https://github.com/getalp/Flaubert) model fine-tuned on [french-translated SQuAD 1.1 dataset](https://github.com/Alikabbadj/French-SQuAD)) + +## Training hyperparameters + +```shell +python3 ./examples/run_squad.py \ +--model_type flaubert \ +--model_name_or_path flaubert-base-uncased \ +--do_train \ +--do_eval \ +--do_lower_case \ +--train_file SQuAD-v1.1-train_fr_ss999_awstart2_net.json \ +--predict_file SQuAD-v1.1-dev_fr_ss999_awstart2_net.json \ +--learning_rate 3e-5 \ +--num_train_epochs 2 \ +--max_seq_length 384 \ +--doc_stride 128 \ +--output_dir output \ +--per_gpu_eval_batch_size=3 \ +--per_gpu_train_batch_size=3 +``` + +## Evaluation results + +```shell +{"f1": 68.66174806561969, "exact_match": 49.299692063176714} +``` + +## Usage + +```python +from transformers import pipeline + +nlp = pipeline('question-answering', model='fmikaelian/flaubert-base-uncased-squad', tokenizer='fmikaelian/flaubert-base-uncased-squad') + +nlp({ + 'question': "Qui est Claude Monet?", + 'context': "Claude Monet, né le 14 novembre 1840 à Paris et mort le 5 décembre 1926 à Giverny, est un peintre français et l’un des fondateurs de l'impressionnisme." +}) +``` \ No newline at end of file