Update README.md
Browse files
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 |
-
#
|
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 |
|
26 |
-
|
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 |
|