Skip to content

Commit

Permalink
added the backward compatibility if the finetuning data is constructe…
Browse files Browse the repository at this point in the history
…d from the OMOP data, where the age_at_index is labelled as age
  • Loading branch information
ChaoPang committed Sep 9, 2024
1 parent 63db5d6 commit 58b761c
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -453,6 +453,6 @@ class HFFineTuningMapping(DatasetMapping):

def transform(self, record: Dict[str, Any]) -> Dict[str, Any]:
return {
"age_at_index": record["age_at_index"],
"age_at_index": record["age"] if "age" in record else record["age_at_index"],
"classifier_label": record["label"],
}

0 comments on commit 58b761c

Please sign in to comment.