Update README.md

README.md

@@ -8,6 +8,9 @@ metrics:
 model-index:
 - name: iati-drr-classifier
   results: []
+datasets:
+- alex-miller/iati-policy-markers
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,14 +18,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # iati-drr-classifier
 
-This model is a fine-tuned version of [alex-miller/ODABert](https://huggingface.co/alex-miller/ODABert) on the None dataset.
+This model is a fine-tuned version of [alex-miller/ODABert](https://huggingface.co/alex-miller/ODABert) on a subset of the [alex-miller/iati-policy-markers](https://huggingface.co/datasets/alex-miller/iati-policy-markers) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.3910
 - Accuracy: 0.8207
 
 ## Model description
 
-More information needed
+This model has been trained to identify disaster risk reduction (DRR) project titles and/or descriptions. It returns "0" for projects with no DRR component, and "1" for projects with DRR as a principal or significant objective.
 
 ## Intended uses & limitations
 
@@ -34,6 +37,8 @@ More information needed
 
 ## Training procedure
 
+Code to subset the dataset and train the model is available [here](https://github.com/akmiller01/iati-policy-marker-hf-dataset/blob/main/use_cases/drr_train.ipynb).
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -60,4 +65,4 @@ The following hyperparameters were used during training:
 - Transformers 4.38.2
 - Pytorch 2.2.1+cu121
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2