d0choa commited on
Commit
e88d51c
1 Parent(s): bbffacc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -24
README.md CHANGED
@@ -18,34 +18,14 @@ library_name: transformers
18
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
19
  should probably proofread and complete it, then remove this comment. -->
20
 
21
- # stop_reasons_classificator_multilabel
22
 
23
- This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the task of classification of why a clinical trial has stopped early in 17 different classes. The datased used for fine tuning was manually curated by a group of experts in Open Targets and is also available for download at the [Hub](https://huggingface.co/datasets/opentargets/clinical_trial_reason_to_stop).
24
 
 
25
 
26
- It achieves the following results on the evaluation set:
27
- - Loss: 0.0899
28
- - Accuracy Thresh: 0.9760
29
 
30
- ## Model description
31
-
32
- This research has been done by Olesya Razuvayevskaya (@LesyaR).
33
-
34
- We fine-tuned BERT model for the task of predicting the stop reasons on the training set of 3,571
35
- human-annotated stopped clinical trials (Devlin et al., 2018). We used a BERT uncased pre-trained
36
- model with a one-layer feed-forward classifier. The fine-tuning was performed by using the
37
- Hugging Face transformer library (Wolf et al., 2019). The classifier uses 50 hidden units and the
38
- ReLu activation function.
39
-
40
- ## Intended uses & limitations
41
-
42
- This model is intended to be used by the whole scientific community. It is Apache 2.0 licensed.
43
-
44
- ## Training and evaluation data
45
-
46
- An expert-curated data set of >5000 reasons why a clinical trials have stopped. These data have been extracted from clinicaltrials.gov.
47
-
48
- A set of experts from the Open Targets Consortium assigned these free text labels to a set of 17 different classes after receiving training.
49
 
50
  ## Training procedure
51
 
 
18
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
19
  should probably proofread and complete it, then remove this comment. -->
20
 
21
+ # clinical_trial_stop_reasons
22
 
23
+ This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the task of classification of why a clinical trial has stopped early.
24
 
25
+ The dataset containing 3,747 manually curated reasons used for fine-tuning is available at [Hub](https://huggingface.co/datasets/opentargets/clinical_trial_reason_to_stop).
26
 
27
+ More details on the model training are available in the github project ([link](https://github.com/opentargets/stopReasons)) and in the associated publication (TBC).
 
 
28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
 
30
  ## Training procedure
31